Spaces:
Menyu
/
Running on Zero

Menyu commited on
Commit
e0f9b66
·
verified ·
1 Parent(s): 57440b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ if torch.cuda.is_available():
18
  use_safetensors=True,
19
  add_watermarker=False
20
  )
21
- pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", prediction_type="sample")
22
  pipe.to("cuda")
23
 
24
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
18
  use_safetensors=True,
19
  add_watermarker=False
20
  )
21
+ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
22
  pipe.to("cuda")
23
 
24
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: