coralLight commited on
Commit
11297fe
·
1 Parent(s): 7532355

add inference

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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, steps=num_inference_steps, guidance_scale=guidance_scale)
 
 
 
 
 
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