hugging face 运行模型Google gemma

最新推荐文章于 2024-07-29 21:16:24 发布

hehui0921

最新推荐文章于 2024-07-29 21:16:24 发布

阅读量382

点赞数 1

分类专栏： huggingface 文章标签： python 前端开发语言

本文链接：https://blog.csdn.net/oHeHui1/article/details/136224018

版权

huggingface 专栏收录该内容

33 篇文章 0 订阅

订阅专栏

!pip install -U transformers
from transformers import AutoTokenizer, AutoModelForCausalLM
from huggingface_hub.hf_api import HfFolder

HfFolder.save_token('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')

tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b")
model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")

input_text = "Write me a poem about Machine Learning."
input_ids = tokenizer(input_text, return_tensors="pt")

outputs = model.generate(**input_ids)
print(tokenizer.decode(outputs[0]))