Update README.md
Browse files
README.md
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
```
|
2 |
#!/usr/bin/env python
|
3 |
import torch
|
4 |
from transformers import (
|
@@ -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 |
|
|
|
1 |
+
```python
|
2 |
#!/usr/bin/env python
|
3 |
import torch
|
4 |
from transformers import (
|
|
|
11 |
|
12 |
def main():
|
13 |
# Define the input and output repository names.
|
14 |
+
input_model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
|
15 |
split_2 = input_model_id.split("/")[1]
|
16 |
output_model_id = f"baseten/example-{split_2}ForSequenceClassification"
|
17 |
|