Geonmo commited on
Commit
495dc6a
·
1 Parent(s): 7025767

fix Blending

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -162,7 +162,7 @@ class GraphitPipeline(StableDiffusionInstructPix2PixPipeline):
162
 
163
  # 2. Encode input prompt
164
  cond_embeds = torch.cat([image_cond_embeds, negative_image_cond_embeds])
165
- cond_embeds = einops.repeat(cond_embeds, 'b n d -> (b num) n d', num=num_images_per_prompt) #.to(torch_dtype)
166
  prompt_embeds = cond_embeds
167
 
168
  # 3. Preprocess image
 
162
 
163
  # 2. Encode input prompt
164
  cond_embeds = torch.cat([image_cond_embeds, negative_image_cond_embeds])
165
+ cond_embeds = einops.repeat(cond_embeds, 'b n d -> (b num) n d', num=num_images_per_prompt).to(model_dict['torch_dtype'])
166
  prompt_embeds = cond_embeds
167
 
168
  # 3. Preprocess image