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