Tibetan Roberta Causal Base

🚀 藏文Roberta因果語言模型文本生成示例

本項目提供了一個使用 藏文Roberta因果語言模型 進行文本生成的示例，藉助該模型可以實現藏文文本的生成，為藏文自然語言處理相關應用提供基礎。

🚀 快速開始

以下是使用 藏文Roberta因果語言模型 進行文本生成的代碼示例：

from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

model_name = 'sangjeedondrub/tibetan-roberta-causal-base'
model = AutoModelForCausalLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

text_gen_pipe = pipeline(
    "text-generation",
    model=model,
    tokenizer=tokenizer
)

init_text = 'རིན་'

outputs = text_gen_pipe(init_text,
              do_sample=True,
              max_new_tokens=200,
              temperature=.9,
              top_k=10,
              top_p=0.92,
              num_return_sequences=10,
              truncate=True)
for idx, output in enumerate(outputs, start=1):
  print(idx)
  print(output['generated_text'])

📚 詳細文檔

模型信息

關於模型

此模型由Sangjee Dondrub [sangjeedondrub at live dot com] 訓練併發布。進行這些實驗的唯一目的是提高其對Transformers API的熟悉程度。

📄 許可證

本項目採用MIT許可證。