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
open-webui
Commits
63ab526b
Unverified
Commit
63ab526b
authored
Jul 14, 2024
by
Timothy Jaeryang Baek
Committed by
GitHub
Jul 14, 2024
Browse files
Merge pull request #3815 from JerryKwan/issue3813
fix the checking logic in get_model_path
parents
cf92909d
f07172bf
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
3 deletions
+6
-3
backend/apps/rag/utils.py
backend/apps/rag/utils.py
+6
-3
No files found.
backend/apps/rag/utils.py
View file @
63ab526b
...
...
@@ -326,22 +326,25 @@ def get_model_path(model: str, update_model: bool = False):
local_files_only
=
not
update_model
local_model_path
=
os
.
path
.
join
(
cache_dir
,
model
)
snapshot_kwargs
=
{
"cache_dir"
:
cache_dir
,
"local_files_only"
:
local_files_only
,
}
log
.
debug
(
f
"model:
{
model
}
"
)
log
.
debug
(
f
"local_model_path:
{
local_model_path
}
"
)
log
.
debug
(
f
"snapshot_kwargs:
{
snapshot_kwargs
}
"
)
# Inspiration from upstream sentence_transformers
if
(
os
.
path
.
exists
(
model
)
os
.
path
.
exists
(
local_model_path
)
or
(
"
\\
"
in
model
or
model
.
count
(
"/"
)
>
1
)
and
local_files_only
):
# If fully qualified path exists, return input, else set repo_id
return
model
return
local_model_path
elif
"/"
not
in
model
:
# Set valid repo_id for model short-name
model
=
"sentence-transformers"
+
"/"
+
model
...
...
@@ -355,7 +358,7 @@ def get_model_path(model: str, update_model: bool = False):
return
model_repo_path
except
Exception
as
e
:
log
.
exception
(
f
"Cannot determine model snapshot path:
{
e
}
"
)
return
model
return
local_model_path
def
generate_openai_embeddings
(
...
...
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