Spaces:
Sleeping
Sleeping
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 |
|