zjunlp
/

OceanGPT-coder-0.6B

Model card Files Files and versions Community

ShawnRu commited on 4 days ago

Commit

c1570a1

·

verified ·

1 Parent(s): 9c0c1bc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ huggingface-cli download --resume-download zjunlp/OceanGPT-coder-0.6B --local-di
 ### Inference
 ```python
-from modelscope import AutoModelForCausalLM, AutoTokenizer
 model_name = "zjunlp/OceanGPT-coder-0.6B"
 device = "auto" # or your divice, e.g., "cuda:0" or "cpu".

 ### Inference
 ```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
 model_name = "zjunlp/OceanGPT-coder-0.6B"
 device = "auto" # or your divice, e.g., "cuda:0" or "cpu".