michaelfeil commited on
Commit
c26ba8d
·
verified ·
1 Parent(s): bb623b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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-70B-Instruct"
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