BAAI
/

ryanzhangfan commited on
Commit
083e245
1 Parent(s): 6244a14

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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)