Spaces:
Paused
Paused
Update app.py
Browse files
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(
|
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,
|