kdf/jiang-base

Model Card for Model ID

This modelcard aims to be a base template for new models. It has been generated using this raw template.

Model Details

Model Description

• Developed by: 知未智能KDF
• Model type: JIANG
• Language(s) (NLP): Chinese
• License: apache-2.0

How to Get Started with the Model

Use the code below to get started with the model.

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = 'kdf/jiang-base'
model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
ret = model.generate(**tokenizer('今天天气不错', return_tensors='pt'), max_new_tokens=50, top_k=1)
print(tokenizer.decode(ret[0]))