Skip to content

GitLab

"mmdet3d/configs/_base_/datasets/scannet_seg.py" did not exist on "9d852f1721d50713f04ac28bf69d60b7458214ab"
Switch branch/tag
  1. 08 Apr, 2025 2 commits
    • TorynCurtis's avatar
      wan model cpu_offload (#3) · b93699b0
      TorynCurtis authored 
      * 修改了main.py, t5的model, wan的model、三个weights文件和三个infer文件, 并且在common的conv3d算子中注册新算子

* 修改了Conv3dWeightForceBF16算子，更新了wan的pre_weights中对此算子的使用

* 修复了import中的bug

* 修复了WanPreWeights, WanTransformerWeights没有self.config的bug

* 修复了WanPreWeights, WanTransformerWeights没有self.config的bug

* 修复了config的bug，目前在使用cpu_offload的时候，vae阶段有tensor不在同一device的bug

* 修复了vae阶段迁移的bug

* 修复了scale在mean和inv_std迁移后仍需重新赋值的bug
      b93699b0
    • helloyongyang's avatar
      first commit · daf4c74e
      helloyongyang authored
      daf4c74e