Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ base_model_path = "runwayml/stable-diffusion-v1-5"
|
|
| 16 |
vae_model_path = "stabilityai/sd-vae-ft-mse"
|
| 17 |
image_encoder_path = "models/image_encoder/"
|
| 18 |
ip_ckpt = "models/ip-adapter_sd15.bin"
|
| 19 |
-
device = "
|
| 20 |
|
| 21 |
# VAE and scheduler
|
| 22 |
noise_scheduler = DDIMScheduler(
|
|
|
|
| 16 |
vae_model_path = "stabilityai/sd-vae-ft-mse"
|
| 17 |
image_encoder_path = "models/image_encoder/"
|
| 18 |
ip_ckpt = "models/ip-adapter_sd15.bin"
|
| 19 |
+
device = "cpu" # or "cuda" if using GPU
|
| 20 |
|
| 21 |
# VAE and scheduler
|
| 22 |
noise_scheduler = DDIMScheduler(
|