PEFT
Safetensors
German
trl
sft
Generated from Trainer
JanPf commited on
Commit
515d480
1 Parent(s): f9a712c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
31
  torch.manual_seed(42)
32
 
33
  # script config
34
- base_model_name = "LSX-UniWue/llammchen_1b"
35
  chat_adapter_name = "LSX-UniWue/LLaMmlein_1B_chat_selected"
36
  device = "mps" # or cuda
37
 
 
31
  torch.manual_seed(42)
32
 
33
  # script config
34
+ base_model_name = "LSX-UniWue/LLaMmlein_1B"
35
  chat_adapter_name = "LSX-UniWue/LLaMmlein_1B_chat_selected"
36
  device = "mps" # or cuda
37