Update app.py
Browse files
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 |
-
|
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
|
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,
|