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
chenpangpang
ComfyUI
Commits
7df82221
Commit
7df82221
authored
Sep 10, 2023
by
comfyanonymous
Browse files
Allow checkpoints with .pt and .bin extensions.
parent
07691e80
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
folder_paths.py
folder_paths.py
+1
-2
No files found.
folder_paths.py
View file @
7df82221
import
os
import
time
supported_ckpt_extensions
=
set
([
'.ckpt'
,
'.pth'
,
'.safetensors'
])
supported_pt_extensions
=
set
([
'.ckpt'
,
'.pt'
,
'.bin'
,
'.pth'
,
'.safetensors'
])
folder_names_and_paths
=
{}
base_path
=
os
.
path
.
dirname
(
os
.
path
.
realpath
(
__file__
))
models_dir
=
os
.
path
.
join
(
base_path
,
"models"
)
folder_names_and_paths
[
"checkpoints"
]
=
([
os
.
path
.
join
(
models_dir
,
"checkpoints"
)],
supported_
ck
pt_extensions
)
folder_names_and_paths
[
"checkpoints"
]
=
([
os
.
path
.
join
(
models_dir
,
"checkpoints"
)],
supported_pt_extensions
)
folder_names_and_paths
[
"configs"
]
=
([
os
.
path
.
join
(
models_dir
,
"configs"
)],
[
".yaml"
])
folder_names_and_paths
[
"loras"
]
=
([
os
.
path
.
join
(
models_dir
,
"loras"
)],
supported_pt_extensions
)
...
...
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