Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
0488810f
Unverified
Commit
0488810f
authored
Nov 13, 2023
by
Nicolas Hug
Committed by
GitHub
Nov 13, 2023
Browse files
Fix realfill example compatibility with latest torchvision version (#5736)
parent
ef7787ea
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
examples/research_projects/realfill/requirements.txt
examples/research_projects/realfill/requirements.txt
+1
-1
examples/research_projects/realfill/train_realfill.py
examples/research_projects/realfill/train_realfill.py
+2
-2
No files found.
examples/research_projects/realfill/requirements.txt
View file @
0488810f
...
@@ -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.1
5.2
torchvision
>
=0.1
6
ftfy==6.1.1
ftfy==6.1.1
tensorboard==2.14.0
tensorboard==2.14.0
Jinja2==3.1.2
Jinja2==3.1.2
examples/research_projects/realfill/train_realfill.py
View file @
0488810f
...
@@ -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
]),
]
]
)
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment