开发者社区 > ModelScope模型即服务 > 正文

ModelScope报错怎么解决?

ModelScope报错怎么解决? :TypeError: init() got an unexpected keyword argument 'device' 03860f5dfabd58d9e774104380f2ae28.png

展开
收起
小小爱吃香菜 2024-02-13 21:14:45 106 0
1 条回答
写回答
取消 提交回答
  • 北京阿里云ACE会长

    创建模型时,为模型的初始化函数传递了不必要的“device”关键字参数。为了解决这个问题,请尝试以下方法:

    1. 确保您已经正确安装了CUDA扩展。如果没有安装,请根据NVIDIA的官方文档进行安装:image.png
      https://docs.nvidia.com/cuda/11.0/install-guide/index.html
    2. 在创建模型时,不要传递“device”参数。将模型的创建代码更改为:

    model = AutoModelForCausalLM.from_pretrained(pretrained_model_name)

    1. 如果您的系统没有CUDA支持,您可以尝试使用PyTorch的“CPUOnly”选项。在创建模型之前,添加以下代码行:

    torch.set_default_tensor_type('torch.FloatTensor')

    这将在创建模型时使用CPU而非GPU。但是请注意,这将导致模型的性能下降。

    1. 如果您确实需要使用GPU,但安装了错误的CUDA版本,请尝试更新或降级CUDA版本以与您的系统兼容。

    image.png

    2024-02-14 09:51:41
    赞同 展开评论 打赏

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载