Fix model name (#1)
Browse files- Fix model name (cddf3c49cfae9778207996478204bb3160016261)
Co-authored-by: Manuel Romero <[email protected]>
README.md
CHANGED
|
@@ -54,7 +54,7 @@ Here's a [Colab notebook](https://colab.research.google.com/drive/1k6C_oJfEKUq0m
|
|
| 54 |
import torch
|
| 55 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 56 |
|
| 57 |
-
model_name = "phixtral-
|
| 58 |
instruction = '''
|
| 59 |
def print_prime(n):
|
| 60 |
"""
|
|
|
|
| 54 |
import torch
|
| 55 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 56 |
|
| 57 |
+
model_name = "phixtral-2x2_8"
|
| 58 |
instruction = '''
|
| 59 |
def print_prime(n):
|
| 60 |
"""
|