Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,12 +30,6 @@ JS = """function () {
|
|
| 30 |
}
|
| 31 |
}"""
|
| 32 |
|
| 33 |
-
# Load VAE component
|
| 34 |
-
vae = AutoencoderKL.from_pretrained(
|
| 35 |
-
vae_model,
|
| 36 |
-
torch_dtype=torch.float16
|
| 37 |
-
)
|
| 38 |
-
|
| 39 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
| 40 |
if torch.cuda.is_available():
|
| 41 |
pipe = StableAudioPipeline.from_pretrained(
|
|
|
|
| 30 |
}
|
| 31 |
}"""
|
| 32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
pipe = StableAudioPipeline.from_pretrained(
|