Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 2, 2023

Commit

4184307

1 Parent(s): f55a34e

ToTensor instead of PILToTensor

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ conditioning_image_transforms = T.Compose(
     [
         #T2.ScaleJitter(target_size=output_res, scale_range=(0.5, 3.0))),
         T.RandomCrop(size=output_res, pad_if_needed=True, padding_mode="symmetric"),
-        T.PILToTensor(),
         T.Normalize([0.5], [0.5]),
     ]
 )

     [
         #T2.ScaleJitter(target_size=output_res, scale_range=(0.5, 3.0))),
         T.RandomCrop(size=output_res, pad_if_needed=True, padding_mode="symmetric"),
+        T.ToTensor(),
         T.Normalize([0.5], [0.5]),
     ]
 )