Update README.md
Browse files
README.md
CHANGED
|
@@ -86,6 +86,7 @@ kwargs = dict(
|
|
| 86 |
ratio="1:1",
|
| 87 |
image_area=model.config.image_area,
|
| 88 |
return_tensors="pt",
|
|
|
|
| 89 |
)
|
| 90 |
pos_inputs = processor(text=prompt, **kwargs)
|
| 91 |
neg_inputs = processor(text=NEGATIVE_PROMPT, **kwargs)
|
|
|
|
| 86 |
ratio="1:1",
|
| 87 |
image_area=model.config.image_area,
|
| 88 |
return_tensors="pt",
|
| 89 |
+
padding="longest",
|
| 90 |
)
|
| 91 |
pos_inputs = processor(text=prompt, **kwargs)
|
| 92 |
neg_inputs = processor(text=NEGATIVE_PROMPT, **kwargs)
|