有朋友体验过ModelScope yi-34b模型吗?

有朋友体验过ModelScope yi-34b模型吗?有没有示例代码

展开
收起
真的很搞笑 2023-11-15 08:02:36 133 分享 版权
1 条回答
写回答
取消 提交回答
  • 是的,ModelScope 提供了 Yi-34b 模型。以下是一个简单的示例代码:

    from transformers import AutoTokenizer, AutoModel
    
    tokenizer = AutoTokenizer.from_pretrained("modelscope/yi-34b")
    model = AutoModel.from_pretrained("modelscope/yi-34b")
    
    inputs = tokenizer("Hello, world!", return_tensors="pt")
    outputs = model(**inputs)
    

    这段代码首先加载了Yi-34b模型和对应的分词器,然后对输入进行了编码。接着,我们使用AutoModel类来创建模型实例,并将输入传递给模型进行推理。

    2023-11-15 10:20:19
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

还有其他疑问?
咨询AI助理