Spaces:
vilarin
/
Running on Zero

vilarin commited on
Commit
d748926
·
verified ·
1 Parent(s): 3bb60df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ if torch.cuda.is_available():
33
  transformer=transformer,
34
  torch_dtype=torch.bfloat16)
35
  pipe.scheduler = EulerDiscreteScheduler.from_config(
36
- pipe.scheduler.config, timestep_spacing="linspace", use_beta_sigmas=True
37
  )
38
  pipe.to("cuda")
39
 
 
33
  transformer=transformer,
34
  torch_dtype=torch.bfloat16)
35
  pipe.scheduler = EulerDiscreteScheduler.from_config(
36
+ pipe.scheduler.config, use_beta_sigmas=True
37
  )
38
  pipe.to("cuda")
39