Menyu commited on
Commit
349a197
·
verified ·
1 Parent(s): b7b00ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -21,16 +21,16 @@ if torch.cuda.is_available():
21
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
22
  pipe.scheduler.register_to_config(
23
  rescale_betas_zero_snr=True,
24
- clip_sample=false,
25
  interpolation_type=linear,
26
  prediction_type=epsilon,
27
  sample_max_value=1.0,
28
- set_alpha_to_one=false,
29
- skip_prk_steps=true,
30
  steps_offset=1,
31
  timestep_spacing=leading,
32
  trained_betas=null,
33
- use_karras_sigmas=false
34
  )
35
  pipe.to("cuda")
36
 
 
21
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
22
  pipe.scheduler.register_to_config(
23
  rescale_betas_zero_snr=True,
24
+ clip_sample=False,
25
  interpolation_type=linear,
26
  prediction_type=epsilon,
27
  sample_max_value=1.0,
28
+ set_alpha_to_one=False,
29
+ skip_prk_steps=True,
30
  steps_offset=1,
31
  timestep_spacing=leading,
32
  trained_betas=null,
33
+ use_karras_sigmas=False
34
  )
35
  pipe.to("cuda")
36