Yuanshi commited on
Commit
8528685
·
verified ·
1 Parent(s): 2045dce

Update ominicontrol.py

Browse files
Files changed (1) hide show
  1. ominicontrol.py +2 -1
ominicontrol.py CHANGED
@@ -63,7 +63,7 @@ def generate_image(
63
  "The Simpsons": "simpsons",
64
  "Snoopy": "snoopy",
65
  }[style]
66
- # pipe.set_adapters(activate_adapter_name)
67
 
68
  factor = 512 / max(image.size)
69
  image = resize(
@@ -132,6 +132,7 @@ def generate_image(
132
  "image_guidance_scale": image_guidance,
133
  "seed": seed,
134
  "steps": steps,
 
135
  "width": width,
136
  "height": height,
137
  }
 
63
  "The Simpsons": "simpsons",
64
  "Snoopy": "snoopy",
65
  }[style]
66
+ pipe.set_adapters(activate_adapter_name)
67
 
68
  factor = 512 / max(image.size)
69
  image = resize(
 
132
  "image_guidance_scale": image_guidance,
133
  "seed": seed,
134
  "steps": steps,
135
+ "style": style,
136
  "width": width,
137
  "height": height,
138
  }