Blazgo commited on
Commit
6b69fab
·
verified ·
1 Parent(s): 4e8cc0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- tmp_env = os.environ.copy()
 
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)