Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,6 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
| 33 |
# Load VAE component
|
| 34 |
vae = AutoencoderKL.from_pretrained(
|
| 35 |
vae_model,
|
| 36 |
-
torch_dtype="auto"
|
| 37 |
)
|
| 38 |
|
| 39 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
|
|
|
| 33 |
# Load VAE component
|
| 34 |
vae = AutoencoderKL.from_pretrained(
|
| 35 |
vae_model,
|
|
|
|
| 36 |
)
|
| 37 |
|
| 38 |
# Ensure model and scheduler are initialized in GPU-enabled function
|