LTT commited on
Commit
fe87c8d
·
verified ·
1 Parent(s): c4a1807

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -104,7 +104,7 @@ flux_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", to
104
  flux_lora_ckpt_path = hf_hub_download(repo_id="LTT/xxx-ckpt", filename="rgb_normal_large.safetensors", repo_type="model")
105
  flux_pipe.load_lora_weights(flux_lora_ckpt_path)
106
  flux_pipe.to(device=device_0, dtype=torch.bfloat16)
107
- # flux_pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(flux_pipe)
108
 
109
 
110
  # lrm
@@ -219,7 +219,7 @@ def generate_multi_view_images(prompt, seed):
219
  # generator=generator,
220
  # good_vae=good_vae,
221
  # ).images
222
- for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
223
  prompt=prompt,
224
  guidance_scale=guidance_scale,
225
  num_inference_steps=num_inference_steps,
 
104
  flux_lora_ckpt_path = hf_hub_download(repo_id="LTT/xxx-ckpt", filename="rgb_normal_large.safetensors", repo_type="model")
105
  flux_pipe.load_lora_weights(flux_lora_ckpt_path)
106
  flux_pipe.to(device=device_0, dtype=torch.bfloat16)
107
+ flux_pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(flux_pipe)
108
 
109
 
110
  # lrm
 
219
  # generator=generator,
220
  # good_vae=good_vae,
221
  # ).images
222
+ for img in flux_pipe.flux_pipe_call_that_returns_an_iterable_of_images(
223
  prompt=prompt,
224
  guidance_scale=guidance_scale,
225
  num_inference_steps=num_inference_steps,