Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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(
|