ori-cloud
/

ds-trinity-7b-v1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

neha-ori commited on Mar 18

Commit

5d4306e

•

1 Parent(s): 404b0df

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ dtype: float16
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-tokenizer = AutoTokenizer.from_pretrained("m-a-p/OpenCodeInterpreter-DS-6.7B")
 model = AutoModelForCausalLM.from_pretrained("ori-cloud/ds-trinity-7b-v1", torch_dtype=torch.bfloat16,
     device_map="auto")
 prompt = "#write a quick sort algorithm"

 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+tokenizer = AutoTokenizer.from_pretrained("ori-cloud/ds-trinity-7b-v1")
 model = AutoModelForCausalLM.from_pretrained("ori-cloud/ds-trinity-7b-v1", torch_dtype=torch.bfloat16,
     device_map="auto")
 prompt = "#write a quick sort algorithm"