当前位置: 首页 > article >正文

pytorch的模型load

torch.load()可以loadpth结尾的训练好的模型类似:

voicecraft_path="/llm/nankai/chenyang_space/projects/VoiceCraft/pretrained_models/giga330M.pth"

ckpt_path = voicecraft_path

ckpt = torch.load(ckpt_path, map_location=device)

model = voicecraft.VoiceCraft(ckpt["config"])

model.load_state_dict(ckpt["model"])

model.to(device)

而 model = voicecraft.VoiceCraft.from_pretrained(f"pyp1/VoiceCraft_{voicecraft_name.replace('.pth', '')}")

这种使用huggingface的from_pretrain的函数的,就需要从huggingface上下载safetensors来跑


http://www.kler.cn/a/400167.html

相关文章:

  • Mac os 系统上如何开启远程登录的 ssh 22 端口
  • candence : 如何利用EXCEL 绘制复杂、多管脚元件
  • 《C++设计模式:重塑游戏角色系统类结构的秘籍》
  • 大数据如何助力干部选拔的公正性
  • 躺平成长-人工智能进行编程-(12)
  • 问:说说SpringDAO及ORM的用法?
  • C语言练习.if.else语句
  • 全新UI H5购物商城系统存在前台任意文件上传漏洞
  • 每日一题3239.最少翻转次数使二进制矩阵回文;
  • Elasticsearch-Elasticsearch-Rest-Client(三)
  • django从入门到实战(一)——路由的编写规则与使用
  • CS DAC的Matlab建模与电路设计
  • MSTP知识点
  • 学习笔记026——Redis基本操作命令
  • CSS(8)高级技巧:精灵图,css三角,用户界面,vertical-align属性应用
  • Golang 不常被注意的特性
  • 使用 Python的Pyecharts 绘制词云图
  • torch.is_storage()
  • 部分利用oracle数据字典查询对应信息的语句。
  • 基于Python的招聘信息推荐系统
  • 力扣(leetcode)面试经典150题——27. 移除元素
  • gitlab容器的迁移(部署)并配置自动备份
  • [webgis 0基础到找工作]------JavaScript--DOM 注册事件 day09
  • vue实现展示并下载后端返回的图片流
  • uniapp 修改网页标题title
  • 【新人系列】Python 入门(十一):控制结构