Fabrice-TIERCELIN commited on
Commit
c9ce776
·
verified ·
1 Parent(s): 66789be

This Pull Request fixes "Error: RuntimeError: Found no NVIDIA driver on your system."

Browse files

As the space is running on CPU, we should specify _CPU_. Working successfully on [this space](https://huggingface.co/spaces/Sovenok-Hacker/Qwen3-TTS).

Click on _Merge_ to add this feature.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def get_model(model_type: str, model_size: str):
35
  model_path = get_model_path(model_type, model_size)
36
  loaded_models[key] = Qwen3TTSModel.from_pretrained(
37
  model_path,
38
- device_map="cuda",
39
  dtype=torch.bfloat16,
40
  token=HF_TOKEN,
41
  # attn_implementation="flash_attention_2",
 
35
  model_path = get_model_path(model_type, model_size)
36
  loaded_models[key] = Qwen3TTSModel.from_pretrained(
37
  model_path,
38
+ device_map="cpu",
39
  dtype=torch.bfloat16,
40
  token=HF_TOKEN,
41
  # attn_implementation="flash_attention_2",