K00B404 commited on
Commit
da14ac2
·
verified ·
1 Parent(s): 6da8d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "cuda" # or "cuda" if using GPU
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(