winglian commited on
Commit
23495a8
1 Parent(s): 91502b9

misc fixes from #943 (#1086) [skip ci]

Browse files
docker/Dockerfile-runpod CHANGED
@@ -5,6 +5,7 @@ ENV HF_DATASETS_CACHE="/workspace/data/huggingface-cache/datasets"
5
  ENV HUGGINGFACE_HUB_CACHE="/workspace/data/huggingface-cache/hub"
6
  ENV TRANSFORMERS_CACHE="/workspace/data/huggingface-cache/hub"
7
  ENV HF_HOME="/workspace/data/huggingface-cache/hub"
 
8
 
9
  COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
10
 
 
5
  ENV HUGGINGFACE_HUB_CACHE="/workspace/data/huggingface-cache/hub"
6
  ENV TRANSFORMERS_CACHE="/workspace/data/huggingface-cache/hub"
7
  ENV HF_HOME="/workspace/data/huggingface-cache/hub"
8
+ ENV HF_HUB_ENABLE_HF_TRANSFER="1"
9
 
10
  COPY scripts/runpod-entrypoint.sh /root/runpod-entrypoint.sh
11
 
src/axolotl/cli/merge_lora.py CHANGED
@@ -35,6 +35,10 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
35
  f"Target directory for merge: `{parsed_cfg.lora_model_dir}` does not exist."
36
  )
37
 
 
 
 
 
38
  do_merge_lora(cfg=parsed_cfg, cli_args=parsed_cli_args)
39
 
40
 
 
35
  f"Target directory for merge: `{parsed_cfg.lora_model_dir}` does not exist."
36
  )
37
 
38
+ parsed_cfg.load_in_4bit = False
39
+ parsed_cfg.load_in_8bit = False
40
+ parsed_cfg.flash_attention = False
41
+
42
  do_merge_lora(cfg=parsed_cfg, cli_args=parsed_cli_args)
43
 
44