From 6a810b16b2ab17445b7178594dac453f2b8c5142 Mon Sep 17 00:00:00 2001 From: Tynan Burke Date: Mon, 17 Apr 2023 18:40:46 -0600 Subject: [PATCH] typo in training.py (#1329) --- modules/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/training.py b/modules/training.py index 3546d9ea..acb2eef8 100644 --- a/modules/training.py +++ b/modules/training.py @@ -149,7 +149,7 @@ def do_train(lora_name: str, always_override: bool, save_steps: int, micro_batch if model_type in MODEL_CLASSES: model_id = MODEL_CLASSES[model_type] else: - model_id == "llama" + model_id = "llama" if model_type == "PeftModelForCausalLM": yield "You are trying to train a LoRA while you already have another LoRA loaded. This will work, but may have unexpected effects. *(Will continue anyway in 5 seconds, press `Interrupt` to stop.)*" print("Warning: Training LoRA over top of another LoRA. May have unexpected effects.")