AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source
This commit is contained in:
Nicolas Hug
2023-11-13 17:55:17 +00:00
committed by GitHub
parent ef7787ea59
commit 0488810f61
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/research_projects/realfill/requirements.txt
View File

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

4
examples/research_projects/realfill/train_realfill.py
View File

@@ -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]),
]
)