multimodalart HF staff commited on
Commit
fa88557
1 Parent(s): 34fa983

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ pipe_edit.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_
34
  pipe_edit.to("cuda")
35
 
36
  pipe_normal = StableDiffusionXLPipeline.from_single_file(normal_file, torch_dtype=torch.float16, vae=vae)
37
- pipe_normal.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction", sigma_schedule="exponential"s)
38
  pipe_normal.to("cuda")
39
 
40
  @spaces.GPU
 
34
  pipe_edit.to("cuda")
35
 
36
  pipe_normal = StableDiffusionXLPipeline.from_single_file(normal_file, torch_dtype=torch.float16, vae=vae)
37
+ pipe_normal.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction", sigma_schedule="exponential")
38
  pipe_normal.to("cuda")
39
 
40
  @spaces.GPU