Spaces:
Running
on
Zero
Running
on
Zero
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"
|