Spaces:
Runtime error
Runtime error
Nupur Kumari
commited on
Commit
Β·
2fbc966
1
Parent(s):
db5f8d8
updates
Browse files- app.py +3 -1
- pipelines/flux_pipeline/pipeline.py +0 -1
app.py
CHANGED
|
@@ -74,6 +74,8 @@ def encode_target_images(images, pipeline):
|
|
| 74 |
|
| 75 |
@spaces.GPU(duration=120)
|
| 76 |
def generate_image(text, img1, img2, img3, guidance_scale, inference_steps, seed, enable_cpu_offload=False, neg_prompt="", true_cfg=1.0, image_cfg=0.0):
|
|
|
|
|
|
|
| 77 |
if enable_cpu_offload:
|
| 78 |
pipeline.enable_sequential_cpu_offload()
|
| 79 |
input_images = [img1, img2, img3]
|
|
@@ -101,7 +103,7 @@ def generate_image(text, img1, img2, img3, guidance_scale, inference_steps, seed
|
|
| 101 |
num_inference_steps=inference_steps,
|
| 102 |
height=512,
|
| 103 |
width=numref * 512,
|
| 104 |
-
generator = torch.Generator(device="
|
| 105 |
joint_attention_kwargs={'shared_attn': True, 'num': numref},
|
| 106 |
return_dict=False,
|
| 107 |
negative_prompt=neg_prompt,
|
|
|
|
| 74 |
|
| 75 |
@spaces.GPU(duration=120)
|
| 76 |
def generate_image(text, img1, img2, img3, guidance_scale, inference_steps, seed, enable_cpu_offload=False, neg_prompt="", true_cfg=1.0, image_cfg=0.0):
|
| 77 |
+
if neg_prompt == "":
|
| 78 |
+
neg_prompt = "3d render, cartoon, low resolution, illustration, blurry, unrealistic"
|
| 79 |
if enable_cpu_offload:
|
| 80 |
pipeline.enable_sequential_cpu_offload()
|
| 81 |
input_images = [img1, img2, img3]
|
|
|
|
| 103 |
num_inference_steps=inference_steps,
|
| 104 |
height=512,
|
| 105 |
width=numref * 512,
|
| 106 |
+
generator = torch.Generator(device="cuda").manual_seed(seed),
|
| 107 |
joint_attention_kwargs={'shared_attn': True, 'num': numref},
|
| 108 |
return_dict=False,
|
| 109 |
negative_prompt=neg_prompt,
|
pipelines/flux_pipeline/pipeline.py
CHANGED
|
@@ -110,7 +110,6 @@ def normalized_guidance_image(neg_noise_pred, noise_pred, image_noise_pred, true
|
|
| 110 |
pred_guided = image_noise_pred + image_cfg_scale * diff_img + true_cfg_scale * diff_txt
|
| 111 |
return pred_guided
|
| 112 |
|
| 113 |
-
|
| 114 |
class SynCDFluxPipeline(FluxPipeline):
|
| 115 |
|
| 116 |
model_cpu_offload_seq = "text_encoder->text_encoder_2->transformer->vae"
|
|
|
|
| 110 |
pred_guided = image_noise_pred + image_cfg_scale * diff_img + true_cfg_scale * diff_txt
|
| 111 |
return pred_guided
|
| 112 |
|
|
|
|
| 113 |
class SynCDFluxPipeline(FluxPipeline):
|
| 114 |
|
| 115 |
model_cpu_offload_seq = "text_encoder->text_encoder_2->transformer->vae"
|