mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-03-11 13:39:50 +00:00
Fixed scaling issue with control images
This commit is contained in:
@@ -247,6 +247,9 @@ class QwenImageEditPlusModel(QwenImageModel):
|
||||
control_img, size=(c_height, c_width), mode="bilinear"
|
||||
)
|
||||
|
||||
# scale to -1 to 1
|
||||
control_img = control_img * 2 - 1
|
||||
|
||||
control_latent = self.encode_images(
|
||||
control_img,
|
||||
device=self.device_torch,
|
||||
|
||||
Reference in New Issue
Block a user