PEFT
Safetensors
German
trl
sft
Generated from Trainer
JanPf commited on
Commit
2354ba6
·
verified ·
1 Parent(s): b002467

Update README.md

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