Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-29 10:59:32 +01:00
Code Issues 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 ( 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
) )