metadata
license: apache-2.0
pipeline_tag: text-generation
tags:
- ocean
- text-generation-inference
language:
- en
💡 Model description
This repo contains a large ocean generative model (OceanGPT) built with ocean science dataset.
🔍 Intended uses
You can use the model to generate responses or input a partial structure for the model to complete.
🛠️ How to use
We wil provide several examples soon and you can modify the input according to your needs.
# Use a pipeline as a high-level helper
>>> from transformers import pipeline
>>> pipe = pipeline("text-generation", model="zjunlp/OceanGPT-7b")
# Load model directly
>>> from transformers import AutoTokenizer, AutoModelForCausalLM
>>> tokenizer = AutoTokenizer.from_pretrained("zjunlp/OceanGPT-7b")
>>> model = AutoModelForCausalLM.from_pretrained("zjunlp/OceanGPT-7b")