Xiaojian9992024 commited on
Commit
9334a41
·
verified ·
1 Parent(s): 694c8c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -239,7 +239,7 @@ def extract(finetuned_model: str, base_model: str, rank: int, hf_token: str, rep
239
  # Set tmp HF_HOME to avoid filling up disk Space
240
  tmp_env = os.environ.copy() # taken from https://stackoverflow.com/a/4453495
241
  tmp_env["HF_HOME"] = f"{tmpdirname}/.cache"
242
- full_cli = f"mergekit-extract-lora {finetuned_model} {base_model} lora --max_rank={rank}"
243
  yield from runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env)
244
 
245
  if runner.exit_code != 0:
 
239
  # Set tmp HF_HOME to avoid filling up disk Space
240
  tmp_env = os.environ.copy() # taken from https://stackoverflow.com/a/4453495
241
  tmp_env["HF_HOME"] = f"{tmpdirname}/.cache"
242
+ full_cli = f"mergekit-extract-lora {finetuned_model} {base_model} lora --max-rank={rank}"
243
  yield from runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env)
244
 
245
  if runner.exit_code != 0: