stardate69 commited on
Commit
4a7abf4
·
verified ·
1 Parent(s): 584d917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 model
20
  pipe = StableAudioPipeline.from_pretrained(
21
  "stabilityai/stable-audio-open-1.0",
22
  torch_dtype=torch_dtype,
23
- use_auth_token=True # Token provided via 'login' earlier
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