baseten
/

example-Meta-Llama-3-70B-InstructForSequenceClassification

Model card Files Files and versions Community

michaelfeil commited on Feb 2

Commit

c26ba8d

·

verified ·

1 Parent(s): bb623b4

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -1,4 +1,4 @@
-```
 #!/usr/bin/env python
 import torch
 from transformers import (
@@ -11,7 +11,7 @@ from transformers import (
 def main():
     # Define the input and output repository names.
-    input_model_id = "meta-llama/Meta-Llama-3-70B-Instruct"
     split_2 = input_model_id.split("/")[1]
     output_model_id = f"baseten/example-{split_2}ForSequenceClassification"

+```python
 #!/usr/bin/env python
 import torch
 from transformers import (
 def main():
     # Define the input and output repository names.
+    input_model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
     split_2 = input_model_id.split("/")[1]
     output_model_id = f"baseten/example-{split_2}ForSequenceClassification"