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

Remove deprecated flag

Browse source
This commit is contained in:
oobabooga 2023-04-15 01:21:19 -03:00
parent 94700cc7a5
commit abef355ed0
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

@ -124,7 +124,7 @@ def do_train(lora_name: str, micro_batch_size: int, batch_size: int, epochs: int
print(f"Warning: LoRA training has only currently been validated for LLaMA models. (Found model type: {model_type})")
time.sleep(5)
if shared.args.wbits > 0 or shared.args.gptq_bits > 0:
if shared.args.wbits > 0:
yield "LoRA training does not yet support 4bit. Please use `--load-in-8bit` for now."
return