Spaces:
Runtime error
Runtime error
disable torchvision v2 transform
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from diffusers import UniPCMultistepScheduler
|
|
4 |
import torch
|
5 |
import torchvision
|
6 |
import torchvision.transforms as T
|
7 |
-
from torchvision.transforms import v2 as T2
|
8 |
import cv2
|
9 |
from PIL import Image
|
10 |
import numpy as np
|
@@ -13,8 +13,8 @@ output_res = (768,768)
|
|
13 |
|
14 |
conditioning_image_transforms = T.Compose(
|
15 |
[
|
16 |
-
T2.ScaleJitter(target_size=output_res, scale_range=(0.5, 3.0)),
|
17 |
-
|
18 |
T.ToTensor(),
|
19 |
T.Normalize([0.5], [0.5]),
|
20 |
]
|
|
|
4 |
import torch
|
5 |
import torchvision
|
6 |
import torchvision.transforms as T
|
7 |
+
#from torchvision.transforms import v2 as T2
|
8 |
import cv2
|
9 |
from PIL import Image
|
10 |
import numpy as np
|
|
|
13 |
|
14 |
conditioning_image_transforms = T.Compose(
|
15 |
[
|
16 |
+
#T2.ScaleJitter(target_size=output_res, scale_range=(0.5, 3.0)),
|
17 |
+
T.RandomCrop(size=output_res, pad_if_needed=True, padding_mode="symmetric"),
|
18 |
T.ToTensor(),
|
19 |
T.Normalize([0.5], [0.5]),
|
20 |
]
|