Commit
·
11297fe
1
Parent(s):
7532355
add inference
Browse files
app.py
CHANGED
|
@@ -181,7 +181,12 @@ def generate_image_with_steps(prompt, negative_prompt, seed, width, height, guid
|
|
| 181 |
if num_inference_steps > 8:
|
| 182 |
sampler = UniPCSampler(pipe,model_closure=model_closure, steps=num_inference_steps, guidance_scale=guidance_scale,skip_type='time_uniform')
|
| 183 |
else:
|
| 184 |
-
sampler = UniPCSampler(pipe,model_closure=model_closure
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 185 |
|
| 186 |
c = prompts
|
| 187 |
uc = ['(worst quality:2), (low quality:2), (normal quality:2), bad anatomy, bad proportions, poorly drawn face, poorly drawn hands, missing fingers, extra limbs, blurry, pixelated, distorted, lowres, jpeg artifacts, watermark, signature, text, (deformed:1.5), (bad hands:1.3), overexposed, underexposed, censored, mutated, extra fingers, cloned face, bad eyes'] * len(c) if guidance_scale != 1.0 else None
|
|
|
|
| 181 |
if num_inference_steps > 8:
|
| 182 |
sampler = UniPCSampler(pipe,model_closure=model_closure, steps=num_inference_steps, guidance_scale=guidance_scale,skip_type='time_uniform')
|
| 183 |
else:
|
| 184 |
+
sampler = UniPCSampler(pipe,model_closure=model_closure
|
| 185 |
+
, steps=num_inference_steps
|
| 186 |
+
, guidance_scale=guidance_scale
|
| 187 |
+
, denoise_to_zero=True
|
| 188 |
+
, need_fp16_discrete_method=True
|
| 189 |
+
, force_not_use_afs=True)
|
| 190 |
|
| 191 |
c = prompts
|
| 192 |
uc = ['(worst quality:2), (low quality:2), (normal quality:2), bad anatomy, bad proportions, poorly drawn face, poorly drawn hands, missing fingers, extra limbs, blurry, pixelated, distorted, lowres, jpeg artifacts, watermark, signature, text, (deformed:1.5), (bad hands:1.3), overexposed, underexposed, censored, mutated, extra fingers, cloned face, bad eyes'] * len(c) if guidance_scale != 1.0 else None
|