Spaces:
Running
on
Zero
Running
on
Zero
Update backend.py
Browse files- flux_app/backend.py +1 -1
flux_app/backend.py
CHANGED
@@ -70,7 +70,7 @@ class ModelManager:
|
|
70 |
good_vae=self.good_vae,
|
71 |
):
|
72 |
yield img
|
73 |
-
|
74 |
def generate_image_to_image(self, prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, lora_scale, seed):
|
75 |
"""Generates an image using image-to-image processing."""
|
76 |
generator = torch.Generator(device=DEVICE).manual_seed(seed)
|
|
|
70 |
good_vae=self.good_vae,
|
71 |
):
|
72 |
yield img
|
73 |
+
#@spaces.GPU(duration=100)
|
74 |
def generate_image_to_image(self, prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, lora_scale, seed):
|
75 |
"""Generates an image using image-to-image processing."""
|
76 |
generator = torch.Generator(device=DEVICE).manual_seed(seed)
|