Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,11 +16,11 @@ login(HUGGINGFACE_TOKEN)
|
|
| 16 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 17 |
torch_dtype = torch.float16 if device == "cuda" else torch.float32
|
| 18 |
|
| 19 |
-
# Load the
|
| 20 |
pipe = StableAudioPipeline.from_pretrained(
|
| 21 |
"stabilityai/stable-audio-open-1.0",
|
| 22 |
torch_dtype=torch_dtype,
|
| 23 |
-
use_auth_token=
|
| 24 |
)
|
| 25 |
pipe = pipe.to(device)
|
| 26 |
|
|
|
|
| 16 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 17 |
torch_dtype = torch.float16 if device == "cuda" else torch.float32
|
| 18 |
|
| 19 |
+
# Load the pipeline
|
| 20 |
pipe = StableAudioPipeline.from_pretrained(
|
| 21 |
"stabilityai/stable-audio-open-1.0",
|
| 22 |
torch_dtype=torch_dtype,
|
| 23 |
+
use_auth_token=HUGGINGFACE_TOKEN # Explicitly use token
|
| 24 |
)
|
| 25 |
pipe = pipe.to(device)
|
| 26 |
|