Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ from transformers import (
|
|
11 |
|
12 |
def main():
|
13 |
# Define the input and output repository names.
|
14 |
-
input_model_id = "meta-llama/Meta-Llama-3-
|
15 |
split_2 = input_model_id.split("/")[1]
|
16 |
output_model_id = f"baseten/example-{split_2}ForSequenceClassification"
|
17 |
|
|
|
11 |
|
12 |
def main():
|
13 |
# Define the input and output repository names.
|
14 |
+
input_model_id = "meta-llama/Meta-Llama-3-70B-Instruct"
|
15 |
split_2 = input_model_id.split("/")[1]
|
16 |
output_model_id = f"baseten/example-{split_2}ForSequenceClassification"
|
17 |
|