KingNish commited on
Commit
db43e18
1 Parent(s): c9195ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def set_timesteps_patched(self, num_inference_steps: int, device = None):
43
  self.sigmas = self.sigmas.to("cpu")
44
 
45
  # Image Editor
46
- edit_file = hf_hub_download(repo, filename="cosxl_edit.safetensors")
47
  EDMEulerScheduler.set_timesteps = set_timesteps_patched
48
  pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_single_file(
49
  edit_file, num_in_channels=8, is_cosxl_edit=True, vae=vae, torch_dtype=torch.float16,
 
43
  self.sigmas = self.sigmas.to("cpu")
44
 
45
  # Image Editor
46
+ edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
47
  EDMEulerScheduler.set_timesteps = set_timesteps_patched
48
  pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_single_file(
49
  edit_file, num_in_channels=8, is_cosxl_edit=True, vae=vae, torch_dtype=torch.float16,