kdf
/

jiang-base

Text Generation Transformers PyTorch Chinese gpt_jiang custom_code

Model card Files Files and versions Community

Edit model card

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]))

Downloads last month: 6