Update README.md
Browse files
README.md
CHANGED
@@ -13,7 +13,7 @@ base_model:
|
|
13 |
- rhysjones/phi-2-orange
|
14 |
---
|
15 |
|
16 |
-
# PhiMiX-2x2B
|
17 |
|
18 |
|
19 |
## Code is work in progress
|
@@ -55,7 +55,7 @@ experts:
|
|
55 |
from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
|
56 |
import torch
|
57 |
|
58 |
-
model_name = "paulilioaica/PhiMiX-2x2B"
|
59 |
|
60 |
torch.set_default_device("cuda")
|
61 |
|
|
|
13 |
- rhysjones/phi-2-orange
|
14 |
---
|
15 |
|
16 |
+
# PhiMiX-2x2B-raw
|
17 |
|
18 |
|
19 |
## Code is work in progress
|
|
|
55 |
from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
|
56 |
import torch
|
57 |
|
58 |
+
model_name = "paulilioaica/PhiMiX-2x2B-raw"
|
59 |
|
60 |
torch.set_default_device("cuda")
|
61 |
|