Nupur Kumari commited on
Commit
2fbc966
·
1 Parent(s): db5f8d8
Files changed (2) hide show
  1. app.py +3 -1
  2. 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="cpu").manual_seed(seed),
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"