Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -183,6 +183,7 @@ def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]
|
|
| 183 |
if not repo_name:
|
| 184 |
yield runner.log("No repo name provided. Generating a random one.")
|
| 185 |
repo_name = f"mergekit-{merge_config.merge_method}"
|
|
|
|
| 186 |
repo_name += "-" + "".join(random.choices(string.ascii_lowercase, k=7))
|
| 187 |
repo_name = repo_name.replace("/", "-").strip("-")
|
| 188 |
|
|
@@ -197,7 +198,8 @@ def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]
|
|
| 197 |
yield runner.log(f"Error creating repo {e}", level="ERROR")
|
| 198 |
return
|
| 199 |
|
| 200 |
-
|
|
|
|
| 201 |
tmp_env["HF_HOME"] = f"{tmpdirname}/.cache"
|
| 202 |
full_cli = cli + f" --lora-merge-cache {tmpdirname}/.lora_cache"
|
| 203 |
yield from runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env)
|
|
|
|
| 183 |
if not repo_name:
|
| 184 |
yield runner.log("No repo name provided. Generating a random one.")
|
| 185 |
repo_name = f"mergekit-{merge_config.merge_method}"
|
| 186 |
+
# Make repo_name "unique" (no need to be extra careful on uniqueness)
|
| 187 |
repo_name += "-" + "".join(random.choices(string.ascii_lowercase, k=7))
|
| 188 |
repo_name = repo_name.replace("/", "-").strip("-")
|
| 189 |
|
|
|
|
| 198 |
yield runner.log(f"Error creating repo {e}", level="ERROR")
|
| 199 |
return
|
| 200 |
|
| 201 |
+
# Set tmp HF_HOME to avoid filling up disk Space
|
| 202 |
+
tmp_env = os.environ.copy() # taken from https://stackoverflow.com/a/4453495
|
| 203 |
tmp_env["HF_HOME"] = f"{tmpdirname}/.cache"
|
| 204 |
full_cli = cli + f" --lora-merge-cache {tmpdirname}/.lora_cache"
|
| 205 |
yield from runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env)
|