dezzman commited on
Commit
2b9a58e
·
verified ·
1 Parent(s): d9e59b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = "CompVis/stable-diffusion-v1-4"
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='CompVis/stable-diffusion-v1-4',
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 'CompVis/stable-diffusion-v1-4'",
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