Spaces:
Sleeping
Sleeping
This Pull Request fixes "Error: RuntimeError: Found no NVIDIA driver on your system."
Browse filesAs 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.
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="
|
| 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",
|