K00B404 commited on
Commit
4e70626
·
verified ·
1 Parent(s): f1c89fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  os.system('git clone https://github.com/tencent-ailab/IP-Adapter.git')
3
  os.system('mv IP-Adapter IP_Adapter')
4
-
5
  import gradio as gr
6
  import torch
7
  from PIL import Image
@@ -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 = "cpu" # or "cuda" if using GPU
20
 
21
  # VAE and scheduler
22
  noise_scheduler = DDIMScheduler(
 
1
  import os
2
  os.system('git clone https://github.com/tencent-ailab/IP-Adapter.git')
3
  os.system('mv IP-Adapter IP_Adapter')
4
+ os.system('ls -a')
5
  import gradio as gr
6
  import torch
7
  from PIL import Image
 
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(