infly
/

OpenCoder-1.5B-Base

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Siming Huang commited on 8 days ago

Commit

cd901de

•

1 Parent(s): 9446d4b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -66,7 +66,7 @@ license_link: https://huggingface.co/infly/OpenCoder-1.5B-Base/blob/main/LICENSE
 ```python
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 model_name = "infly/OpenCoder-1.5B-Base"
 model = AutoModelForCausalLM.from_pretrained(model_name,

 ```python
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
 model_name = "infly/OpenCoder-1.5B-Base"
 model = AutoModelForCausalLM.from_pretrained(model_name,