Cognomen commited on
Commit
859c26b
Β·
1 Parent(s): 2972437

disable torchvision v2 transform

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- T2.RandomCrop(size=output_res, pad_if_needed=True, padding_mode="symmetric"),
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
  ]