Spaces:
Runtime error
Runtime error
fix Blending
Browse files
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)
|
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
|