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
transformers==4.34.0
peft==0.5.0
torch==2.0.1
torchvision==0.15.2
torchvision>=0.16
ftfy==6.1.1
tensorboard==2.14.0
Jinja2==3.1.2
......@@ -450,10 +450,10 @@ class RealFillDataset(Dataset):
self.transform = transforms_v2.Compose(
[
transforms_v2.ToImage(),
transforms_v2.RandomResize(size, int(1.125 * size)),
transforms_v2.RandomCrop(size),
transforms_v2.ToImageTensor(),
transforms_v2.ConvertImageDtype(),
transforms_v2.ToDtype(torch.float32, scale=True),
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