Upload README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 31 |
torch_dtype="auto",
|
| 32 |
device_map="auto"
|
| 33 |
)
|
| 34 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 35 |
|
| 36 |
prompt = "Which is the largest ocean in the world?"
|
| 37 |
messages = [
|
|
|
|
| 31 |
torch_dtype="auto",
|
| 32 |
device_map="auto"
|
| 33 |
)
|
| 34 |
+
tokenizer = AutoTokenizer.from_pretrained("zjunlp/OceanGPT-14B-v0.1")
|
| 35 |
|
| 36 |
prompt = "Which is the largest ocean in the world?"
|
| 37 |
messages = [
|