misc fixes

This commit is contained in:
Wing Lian
2023-07-17 03:00:27 -04:00
parent 02224668c3
commit d75adb9835
3 changed files with 6 additions and 1 deletions

View File

@@ -23,7 +23,8 @@ RUN cd axolotl && \
fi
# fix so that git fetch/pull from remote works
RUN git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
RUN cd axolotl && \
git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*" && \
git config --get remote.origin.fetch
# helper for huggingface-login cli

View File

@@ -35,6 +35,7 @@ LOG = logging.getLogger("axolotl.scripts")
DEFAULT_DATASET_PREPARED_PATH = "last_run_prepared"
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
def choose_device(cfg):

View File

@@ -22,6 +22,9 @@ DEFAULT_LOGGING_CONFIG: Dict[str, Any] = {
},
},
"root": {"handlers": ["console"], "level": os.getenv("LOG_LEVEL", "INFO")},
"loggers": {
"axolotl": {"handlers": ["console"], "level": "DEBUG", "propagate": False},
},
}