Spaces:
Runtime error
Runtime error
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 |
|