Cognomen commited on
Commit
2da962c
Β·
1 Parent(s): 5fd18e6

update torchvision to 0.15.1

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,7 +13,7 @@ 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
  T.RandomCrop(size=output_res, pad_if_needed=True, padding_mode="symmetric"),
18
  T.ToTensor(),
19
  T.Normalize([0.5], [0.5]),
 
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]),