Hugging Face's logo

Spaces:
multimodalart
/
cosxl
Running on Zero

App Files Community
3
multimodalart HF staff commited on
Commit
8f183ed
•
1 Parent(s): 682806c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def resize_image(image, resolution):
41
  EDMEulerScheduler.set_timesteps = set_timesteps_patched
42
 
43
  pipe_edit = CosStableDiffusionXLInstructPix2PixPipeline.from_single_file(
44
- edit_file, num_in_channels=8
45
  )
46
  pipe_edit.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction")
47
  pipe_edit.to("cuda")
 
41
  EDMEulerScheduler.set_timesteps = set_timesteps_patched
42
 
43
  pipe_edit = CosStableDiffusionXLInstructPix2PixPipeline.from_single_file(
44
+ edit_file, num_in_channels=8, torch_dtype=torch.float16
45
  )
46
  pipe_edit.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction")
47
  pipe_edit.to("cuda")