Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ IP_ADAPTER = 'h94/IP-Adapter'
|
|
17 |
IP_ADAPTER_WEIGHT_NAME = "ip-adapter-plus_sd15.bin"
|
18 |
|
19 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
20 |
-
model_id_default = "
|
21 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
22 |
|
23 |
class PipelineManager:
|
@@ -116,7 +116,7 @@ def infer(
|
|
116 |
width=512,
|
117 |
height=512,
|
118 |
num_inference_steps=20,
|
119 |
-
model_id='
|
120 |
seed=42,
|
121 |
guidance_scale=7.0,
|
122 |
lora_scale=0.5,
|
@@ -197,7 +197,7 @@ with gr.Blocks(css=css) as demo:
|
|
197 |
model_id = gr.Textbox(
|
198 |
label="Model ID",
|
199 |
max_lines=1,
|
200 |
-
placeholder="Enter model id like '
|
201 |
value=model_id_default
|
202 |
)
|
203 |
|
|
|
17 |
IP_ADAPTER_WEIGHT_NAME = "ip-adapter-plus_sd15.bin"
|
18 |
|
19 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
20 |
+
model_id_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
21 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
22 |
|
23 |
class PipelineManager:
|
|
|
116 |
width=512,
|
117 |
height=512,
|
118 |
num_inference_steps=20,
|
119 |
+
model_id='stable-diffusion-v1-5/stable-diffusion-v1-5',
|
120 |
seed=42,
|
121 |
guidance_scale=7.0,
|
122 |
lora_scale=0.5,
|
|
|
197 |
model_id = gr.Textbox(
|
198 |
label="Model ID",
|
199 |
max_lines=1,
|
200 |
+
placeholder="Enter model id like 'stable-diffusion-v1-5/stable-diffusion-v1-5'",
|
201 |
value=model_id_default
|
202 |
)
|
203 |
|