Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
907a88b2
Commit
907a88b2
authored
Oct 10, 2022
by
alg-wiki
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Added .webp .bmp
parent
f0ab972f
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
dataset.py
modules/textual_inversion/dataset.py
+1
-1
preprocess.py
modules/textual_inversion/preprocess.py
+1
-1
textual_inversion.py
modules/textual_inversion/textual_inversion.py
+1
-1
No files found.
modules/textual_inversion/dataset.py
View file @
907a88b2
...
@@ -22,7 +22,7 @@ class PersonalizedBase(Dataset):
...
@@ -22,7 +22,7 @@ class PersonalizedBase(Dataset):
self
.
width
=
width
self
.
width
=
width
self
.
height
=
height
self
.
height
=
height
self
.
flip
=
transforms
.
RandomHorizontalFlip
(
p
=
flip_p
)
self
.
flip
=
transforms
.
RandomHorizontalFlip
(
p
=
flip_p
)
self
.
extns
=
[
".jpg"
,
".jpeg"
,
".png"
]
self
.
extns
=
[
".jpg"
,
".jpeg"
,
".png"
,
".webp"
,
".bmp"
]
self
.
dataset
=
[]
self
.
dataset
=
[]
...
...
modules/textual_inversion/preprocess.py
View file @
907a88b2
...
@@ -12,7 +12,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
...
@@ -12,7 +12,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
height
=
process_height
height
=
process_height
src
=
os
.
path
.
abspath
(
process_src
)
src
=
os
.
path
.
abspath
(
process_src
)
dst
=
os
.
path
.
abspath
(
process_dst
)
dst
=
os
.
path
.
abspath
(
process_dst
)
extns
=
[
".jpg"
,
".jpeg"
,
".png"
]
extns
=
[
".jpg"
,
".jpeg"
,
".png"
,
".webp"
,
".bmp"
]
assert
src
!=
dst
,
'same directory specified as source and destination'
assert
src
!=
dst
,
'same directory specified as source and destination'
...
...
modules/textual_inversion/textual_inversion.py
View file @
907a88b2
...
@@ -161,7 +161,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
...
@@ -161,7 +161,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
shared
.
state
.
textinfo
=
"Initializing textual inversion training..."
shared
.
state
.
textinfo
=
"Initializing textual inversion training..."
shared
.
state
.
job_count
=
steps
shared
.
state
.
job_count
=
steps
extns
=
[
".jpg"
,
".jpeg"
,
".png"
]
extns
=
[
".jpg"
,
".jpeg"
,
".png"
,
".webp"
,
".bmp"
]
filename
=
os
.
path
.
join
(
shared
.
cmd_opts
.
embeddings_dir
,
f
'{embedding_name}.pt'
)
filename
=
os
.
path
.
join
(
shared
.
cmd_opts
.
embeddings_dir
,
f
'{embedding_name}.pt'
)
...
...
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