K00B404 commited on
Commit
5e0b0a9
·
verified ·
1 Parent(s): 29d319f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ vae_model_path = "stabilityai/sd-vae-ft-mse"
18
  image_encoder_repo="InvokeAI/ip_adapter_sd_image_encoder"
19
  image_encoder_path = "IP_Adapter/ip_adapter/models/image_encoder/"
20
  ip_ckpt = "ip-adapter_sd15.bin"
21
- device = "cpu" # or "cuda" if using GPU
22
 
23
  # VAE and scheduler
24
  noise_scheduler = DDIMScheduler(
 
18
  image_encoder_repo="InvokeAI/ip_adapter_sd_image_encoder"
19
  image_encoder_path = "IP_Adapter/ip_adapter/models/image_encoder/"
20
  ip_ckpt = "ip-adapter_sd15.bin"
21
+ device = "cuda" # or "cuda" if using GPU
22
 
23
  # VAE and scheduler
24
  noise_scheduler = DDIMScheduler(