Update app.py
Browse files
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 --
|
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:
|