Spaces:
Running
on
Zero
Running
on
Zero
rm os.path.join
Browse files- scripts/generate_prompt.py +1 -1
- scripts/process_utils.py +1 -1
scripts/generate_prompt.py
CHANGED
@@ -52,7 +52,7 @@ def load_wd14_tagger_model():
|
|
52 |
repo_id,
|
53 |
file,
|
54 |
subfolder=SUB_DIR,
|
55 |
-
cache_dir=
|
56 |
force_download=True,
|
57 |
force_filename=file,
|
58 |
)
|
|
|
52 |
repo_id,
|
53 |
file,
|
54 |
subfolder=SUB_DIR,
|
55 |
+
cache_dir=model_dir + "/" + SUB_DIR,
|
56 |
force_download=True,
|
57 |
force_filename=file,
|
58 |
)
|
scripts/process_utils.py
CHANGED
@@ -24,7 +24,7 @@ refine_gen_pipe = None
|
|
24 |
|
25 |
def get_file_path(filename, subfolder):
|
26 |
if use_local:
|
27 |
-
return
|
28 |
else:
|
29 |
return download_file(filename, subfolder)
|
30 |
|
|
|
24 |
|
25 |
def get_file_path(filename, subfolder):
|
26 |
if use_local:
|
27 |
+
return subfolder + "/" + filename
|
28 |
else:
|
29 |
return download_file(filename, subfolder)
|
30 |
|