tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

misc fixes

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/finetune.py
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):