mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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 ( from peft import (
LoraConfig, LoraConfig,
get_peft_model, get_peft_model,
prepare_model_for_kbit_training, prepare_model_for_int8_training,
set_peft_model_state_dict set_peft_model_state_dict
) )