oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-21 16:10:19 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update training.py

Browse source
This commit is contained in:
oobabooga 2023-06-25 12:13:15 -03:00 committed by GitHub
parent 1f5ea451c9
commit 95212edf1f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/training.py
View file

@ -14,7 +14,7 @@ from datasets import Dataset, load_dataset
from peft import (
LoraConfig,
get_peft_model,
prepare_model_for_kbit_training,
prepare_model_for_int8_training,
set_peft_model_state_dict
)