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 |
]
|