Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ from elastic_models.transformers import AutoModelForCausalLM
|
|
| 59 |
|
| 60 |
# Currently we require to have your HF token
|
| 61 |
# as we use original weights for part of layers and
|
| 62 |
-
# model
|
| 63 |
model_name = "meta-llama/Llama-3.2-1B-Instruct"
|
| 64 |
hf_token = ''
|
| 65 |
device = torch.device("cuda")
|
|
|
|
| 59 |
|
| 60 |
# Currently we require to have your HF token
|
| 61 |
# as we use original weights for part of layers and
|
| 62 |
+
# model configuration as well
|
| 63 |
model_name = "meta-llama/Llama-3.2-1B-Instruct"
|
| 64 |
hf_token = ''
|
| 65 |
device = torch.device("cuda")
|