Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-23 00:18:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

typo in training.py (#1329)

Browse Source
This commit is contained in:
Tynan Burke 2023-04-17 18:40:46 -06:00 committed by GitHub
parent ac2973ffc6
commit 6a810b16b2
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

@ -149,7 +149,7 @@ def do_train(lora_name: str, always_override: bool, save_steps: int, micro_batch
if model_type in MODEL_CLASSES: if model_type in MODEL_CLASSES:
model_id = MODEL_CLASSES[model_type] model_id = MODEL_CLASSES[model_type]
else: else:
model_id == "llama" model_id = "llama"
if model_type == "PeftModelForCausalLM": 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.)*" 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.") print("Warning: Training LoRA over top of another LoRA. May have unexpected effects.")