Update README.md
Browse files
README.md
CHANGED
|
@@ -57,7 +57,7 @@ from elastic_models.transformers import AutoModelForCausalLM
|
|
| 57 |
|
| 58 |
# Currently we require to have your HF token
|
| 59 |
# as we use original weights for part of layers and
|
| 60 |
-
# model
|
| 61 |
model_name = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
|
| 62 |
hf_token = ''
|
| 63 |
device = torch.device("cuda")
|
|
|
|
| 57 |
|
| 58 |
# Currently we require to have your HF token
|
| 59 |
# as we use original weights for part of layers and
|
| 60 |
+
# model configuration as well
|
| 61 |
model_name = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
|
| 62 |
hf_token = ''
|
| 63 |
device = torch.device("cuda")
|