Unverified Commit 0488810f authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix realfill example compatibility with latest torchvision version (#5736)

parent ef7787ea
...@@ -3,7 +3,7 @@ accelerate==0.23.0 ...@@ -3,7 +3,7 @@ accelerate==0.23.0
transformers==4.34.0 transformers==4.34.0
peft==0.5.0 peft==0.5.0
torch==2.0.1 torch==2.0.1
torchvision==0.15.2 torchvision>=0.16
ftfy==6.1.1 ftfy==6.1.1
tensorboard==2.14.0 tensorboard==2.14.0
Jinja2==3.1.2 Jinja2==3.1.2
...@@ -450,10 +450,10 @@ class RealFillDataset(Dataset): ...@@ -450,10 +450,10 @@ class RealFillDataset(Dataset):
self.transform = transforms_v2.Compose( self.transform = transforms_v2.Compose(
[ [
transforms_v2.ToImage(),
transforms_v2.RandomResize(size, int(1.125 * size)), transforms_v2.RandomResize(size, int(1.125 * size)),
transforms_v2.RandomCrop(size), transforms_v2.RandomCrop(size),
transforms_v2.ToImageTensor(), transforms_v2.ToDtype(torch.float32, scale=True),
transforms_v2.ConvertImageDtype(),
transforms_v2.Normalize([0.5], [0.5]), transforms_v2.Normalize([0.5], [0.5]),
] ]
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment