Spaces:
Sleeping
Sleeping
Update anime_model.py
Browse files- 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=
|
| 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,
|