mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Change a warning message

Browse source
This commit is contained in:
oobabooga 2023-05-29 22:39:23 -03:00 • committed by GitHub
parent e763ace593
commit 3a6e194bc7
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

@ -234,7 +234,7 @@ def do_train(lora_name: str, always_override: bool, save_steps: int, micro_batch
time.sleep(5) time.sleep(5)
if shared.args.wbits > 0 and not shared.args.monkey_patch: if shared.args.wbits > 0 and not shared.args.monkey_patch:
yield "LoRA training in 4-bit requires loading with `--monkey-patch`" yield "LoRA training with GPTQ models requires loading with `--monkey-patch`"
return return
elif not shared.args.load_in_8bit and shared.args.wbits <= 0: elif not shared.args.load_in_8bit and shared.args.wbits <= 0: