Spaces:
Running
on
Zero
Running
on
Zero
updates
Browse files- Flux-LoRA-Generation-Advanced.zip +0 -3
- flux_app/backend.py +4 -2
Flux-LoRA-Generation-Advanced.zip
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:2537b224b8b98c72939afaf580fd85d9c375d1f6e1f94b1c2f630f22fc0f03ce
|
3 |
-
size 26901
|
|
|
|
|
|
|
|
flux_app/backend.py
CHANGED
@@ -40,8 +40,10 @@ class ModelManager:
|
|
40 |
torch_dtype=DTYPE
|
41 |
)
|
42 |
|
43 |
-
setattr(self.pipe, "flux_pipe_call_that_returns_an_iterable_of_images",
|
44 |
-
|
|
|
|
|
45 |
|
46 |
def generate_image(self, prompt_mash, steps, seed, cfg_scale, width, height, lora_scale):
|
47 |
"""Generates an image using the text-to-image pipeline."""
|
|
|
40 |
torch_dtype=DTYPE
|
41 |
)
|
42 |
|
43 |
+
setattr(self.pipe, "flux_pipe_call_that_returns_an_iterable_of_images", self.process_images)
|
44 |
+
|
45 |
+
def process_images(self, *args, **kwargs):
|
46 |
+
return flux_pipe_call_that_returns_an_iterable_of_images(self.pipe, *args, **kwargs)
|
47 |
|
48 |
def generate_image(self, prompt_mash, steps, seed, cfg_scale, width, height, lora_scale):
|
49 |
"""Generates an image using the text-to-image pipeline."""
|