Remove deprecated flag

This commit is contained in:
oobabooga 2023-04-15 01:21:19 -03:00
parent 94700cc7a5
commit abef355ed0

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})") print(f"Warning: LoRA training has only currently been validated for LLaMA models. (Found model type: {model_type})")
time.sleep(5) 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." yield "LoRA training does not yet support 4bit. Please use `--load-in-8bit` for now."
return return