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 |
|