vilarin commited on
Commit
c530610
1 Parent(s): 50126e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -37,12 +37,10 @@ a T5-based text embedding for text conditioning, and a transformer-based diffusi
37
  </center>
38
  """
39
 
40
- # Ensure model and scheduler are initialized in GPU-enabled function
41
- if torch.cuda.is_available():
42
- pipe = StableAudioPipeline.from_pretrained(
43
- model,
44
- torch_dtype=torch.float16)
45
- pipe = pipe.to("cuda")
46
 
47
 
48
  # Function
 
37
  </center>
38
  """
39
 
40
+ pipe = StableAudioPipeline.from_pretrained(
41
+ model,
42
+ torch_dtype=torch.float16)
43
+ pipe = pipe.to("cuda")
 
 
44
 
45
 
46
  # Function