p1atdev commited on
Commit
056e304
1 Parent(s): cf72c4b

fix: weird typo

Browse files
Files changed (1) hide show
  1. diffusion.py +4 -6
diffusion.py CHANGED
@@ -31,11 +31,9 @@ class ImageGenerator:
31
  use_safetensors=True,
32
  add_watermarker=False,
33
  )
34
- self.pipe.bad_punct_regexscheduler = (
35
- EulerAncestralDiscreteScheduler.from_pretrained(
36
- model_name,
37
- subfolder="scheduler",
38
- )
39
  )
40
 
41
  # xformers
@@ -68,4 +66,4 @@ class ImageGenerator:
68
  width=width,
69
  num_inference_steps=num_inference_steps,
70
  guidance_scale=guidance_scale,
71
- ).images[0]
 
31
  use_safetensors=True,
32
  add_watermarker=False,
33
  )
34
+ self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_pretrained(
35
+ model_name,
36
+ subfolder="scheduler",
 
 
37
  )
38
 
39
  # xformers
 
66
  width=width,
67
  num_inference_steps=num_inference_steps,
68
  guidance_scale=guidance_scale,
69
+ ).images