Spaces:
Running
on
A100
Running
on
A100
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)
|