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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -21,6 +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
  )
25
  pipe.to("cuda")
26
 
 
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