broyang commited on
Commit
16cd030
·
verified ·
1 Parent(s): 96ec5fa

Update anime_model.py

Browse files
Files changed (1) hide show
  1. anime_model.py +1 -6
anime_model.py CHANGED
@@ -172,18 +172,13 @@ class Model:
172
  model = NormalBaeDetector.from_pretrained("lllyasviel/Annotators").to("cuda")
173
  torch.cuda.empty_cache()
174
  gc.collect()
175
- control_image = self.preprocess(
176
- image=image,
177
- image_resolution=image_resolution,
178
- detect_resolution=preprocess_resolution,
179
- )
180
  if prompt == "":
181
  prompt = "anime girl"
182
  print(prompt)
183
  return self.run_pipe(
184
  prompt=self.get_prompt("Hentai Nude Anime Titties of " + prompt, additional_prompt),
185
  negative_prompt=negative_prompt,
186
- control_image=control_image,
187
  num_images=num_images,
188
  num_steps=num_steps,
189
  guidance_scale=guidance_scale,
 
172
  model = NormalBaeDetector.from_pretrained("lllyasviel/Annotators").to("cuda")
173
  torch.cuda.empty_cache()
174
  gc.collect()
 
 
 
 
 
175
  if prompt == "":
176
  prompt = "anime girl"
177
  print(prompt)
178
  return self.run_pipe(
179
  prompt=self.get_prompt("Hentai Nude Anime Titties of " + prompt, additional_prompt),
180
  negative_prompt=negative_prompt,
181
+ control_image=image,
182
  num_images=num_images,
183
  num_steps=num_steps,
184
  guidance_scale=guidance_scale,