zjunlp
/

OceanGPT-14B-v0.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

xyd123 commited on Jul 4

Commit

443d650

•

1 Parent(s): 0a46426

Upload README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -24,11 +24,12 @@ We wil provide several examples soon and you can modify the input according to y
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
     "zjunlp/OceanGPT-14B-v0.1",
-    torch_dtype="auto",
     device_map="auto"
 )
 tokenizer = AutoTokenizer.from_pretrained("zjunlp/OceanGPT-14B-v0.1")

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
     "zjunlp/OceanGPT-14B-v0.1",
+    torch_dtype=torch.bfloat16,
     device_map="auto"
 )
 tokenizer = AutoTokenizer.from_pretrained("zjunlp/OceanGPT-14B-v0.1")