Update README.md
Browse files
README.md
CHANGED
|
@@ -76,7 +76,7 @@ uni_emb = uni_emb.unsqueeze(1)
|
|
| 76 |
print("Extracted UNI:", uni_emb.shape)
|
| 77 |
|
| 78 |
# Get unconditional embedding for classifier-free guidance
|
| 79 |
-
uncond =
|
| 80 |
# Generate new samples using the given mask
|
| 81 |
-
samples =
|
| 82 |
```
|
|
|
|
| 76 |
print("Extracted UNI:", uni_emb.shape)
|
| 77 |
|
| 78 |
# Get unconditional embedding for classifier-free guidance
|
| 79 |
+
uncond = pipeline.get_unconditional_embedding(uni_emb.shape[0])
|
| 80 |
# Generate new samples using the given mask
|
| 81 |
+
samples = pipeline(uni_embeds=uni_emb, controlnet_input=mask, negative_uni_embeds=uncond, guidance_scale=2.5, num_images_per_prompt=1).images
|
| 82 |
```
|