Maria commited on
Commit
bff87b0
·
1 Parent(s): 2387a4c
Files changed (2) hide show
  1. app.py +1 -1
  2. infer.py +3 -3
app.py CHANGED
@@ -26,7 +26,7 @@ def on_checkbox_change(use_advanced):
26
 
27
  with gr.Blocks(css=css) as demo:
28
  with gr.Column(elem_id="col-container"):
29
- gr.Markdown(" # Maria Lashina Text-to-Image Rat Stickers Generation App")
30
 
31
  MODEL_LIST = [
32
  "CompVis/stable-diffusion-v1-4",
 
26
 
27
  with gr.Blocks(css=css) as demo:
28
  with gr.Column(elem_id="col-container"):
29
+ gr.Markdown(" # Maria Lashina T2I Rat Stickers Generation App")
30
 
31
  MODEL_LIST = [
32
  "CompVis/stable-diffusion-v1-4",
infer.py CHANGED
@@ -74,7 +74,7 @@ def get_pipe(
74
  weight_name="ip-adapter-plus_sd14.bin",
75
  )
76
 
77
- elif controlnet:
78
 
79
  print('Pipe with ControlNet')
80
 
@@ -90,7 +90,7 @@ def get_pipe(
90
  safety_checker=None,
91
  ).to(device)
92
 
93
- elif ip_adapter:
94
 
95
  print('Pipe with IpAdapter')
96
 
@@ -105,7 +105,7 @@ def get_pipe(
105
  subfolder="models",
106
  weight_name="ip-adapter-plus_sd14.bin")
107
 
108
- else:
109
 
110
  print('Pipe with only SD')
111
 
 
74
  weight_name="ip-adapter-plus_sd14.bin",
75
  )
76
 
77
+ elif use_controlnet and not use_ip_adapter:
78
 
79
  print('Pipe with ControlNet')
80
 
 
90
  safety_checker=None,
91
  ).to(device)
92
 
93
+ elif use_ip_adapter and not use_controlnetconto:
94
 
95
  print('Pipe with IpAdapter')
96
 
 
105
  subfolder="models",
106
  weight_name="ip-adapter-plus_sd14.bin")
107
 
108
+ elif not use_controlnet and not use_ip_adapter:
109
 
110
  print('Pipe with only SD')
111