Update generator.py
Browse files- generator.py +2 -2
generator.py
CHANGED
|
@@ -9,7 +9,7 @@ class Generator:
|
|
| 9 |
self,
|
| 10 |
user_prompt,
|
| 11 |
negative_prompt="",
|
| 12 |
-
guidance_scale=1.
|
| 13 |
num_inference_steps=8,
|
| 14 |
seed=-1
|
| 15 |
):
|
|
@@ -36,7 +36,7 @@ class Generator:
|
|
| 36 |
generator=generator,
|
| 37 |
num_inference_steps=num_inference_steps,
|
| 38 |
guidance_scale=guidance_scale,
|
| 39 |
-
clip_skip=
|
| 40 |
).images[0]
|
| 41 |
|
| 42 |
return result
|
|
|
|
| 9 |
self,
|
| 10 |
user_prompt,
|
| 11 |
negative_prompt="",
|
| 12 |
+
guidance_scale=1.6,
|
| 13 |
num_inference_steps=8,
|
| 14 |
seed=-1
|
| 15 |
):
|
|
|
|
| 36 |
generator=generator,
|
| 37 |
num_inference_steps=num_inference_steps,
|
| 38 |
guidance_scale=guidance_scale,
|
| 39 |
+
clip_skip=1, # Optional, often helps with anime/pixel styles
|
| 40 |
).images[0]
|
| 41 |
|
| 42 |
return result
|