misc fixes
Browse files- docker/Dockerfile +2 -1
- scripts/finetune.py +1 -0
- src/axolotl/logging_config.py +3 -0
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
|
|
|
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 |
|