This commit is contained in:
Daniel Garvey
2023-08-04 14:54:33 -05:00
committed by GitHub
parent a686d7d89f
commit a57eccc997
2 changed files with 8 additions and 2 deletions

View File

@@ -136,7 +136,10 @@ class LoraDataset(Dataset):
if self.center_crop:
crop = min(img.shape[0], img.shape[1])
(h, w,) = (
(
h,
w,
) = (
img.shape[0],
img.shape[1],
)