Update README.md
Browse files
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/
|
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 |
|