ruslanmv commited on
Commit
9f35428
·
1 Parent(s): 8236b2b
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
- lambda *args, **kwargs: flux_pipe_call_that_returns_an_iterable_of_images(self.pipe, *args, **kwargs))
 
 
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."""