PEFT
Safetensors
German
trl
sft
Generated from Trainer
JanPf commited on
Commit
cbadb31
·
verified ·
1 Parent(s): 8ba72f5

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_sharegpt"
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_sharegpt"
34
  device = "mps" # or cuda
35