winglian commited on
Commit
d75adb9
1 Parent(s): 0222466

misc fixes

Browse files
docker/Dockerfile CHANGED
@@ -23,7 +23,8 @@ RUN cd axolotl && \
23
  fi
24
 
25
  # fix so that git fetch/pull from remote works
26
- RUN git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
 
27
  git config --get remote.origin.fetch
28
 
29
  # helper for huggingface-login cli
 
23
  fi
24
 
25
  # fix so that git fetch/pull from remote works
26
+ RUN cd axolotl && \
27
+ git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
28
  git config --get remote.origin.fetch
29
 
30
  # helper for huggingface-login cli
scripts/finetune.py CHANGED
@@ -35,6 +35,7 @@ LOG = logging.getLogger("axolotl.scripts")
35
 
36
 
37
  DEFAULT_DATASET_PREPARED_PATH = "last_run_prepared"
 
38
 
39
 
40
  def choose_device(cfg):
 
35
 
36
 
37
  DEFAULT_DATASET_PREPARED_PATH = "last_run_prepared"
38
+ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
39
 
40
 
41
  def choose_device(cfg):
src/axolotl/logging_config.py CHANGED
@@ -22,6 +22,9 @@ DEFAULT_LOGGING_CONFIG: Dict[str, Any] = {
22
  },
23
  },
24
  "root": {"handlers": ["console"], "level": os.getenv("LOG_LEVEL", "INFO")},
 
 
 
25
  }
26
 
27
 
 
22
  },
23
  },
24
  "root": {"handlers": ["console"], "level": os.getenv("LOG_LEVEL", "INFO")},
25
+ "loggers": {
26
+ "axolotl": {"handlers": ["console"], "level": "DEBUG", "propagate": False},
27
+ },
28
  }
29
 
30