fix import error on new comfyui portable

FuouM · Aug 17, 2024 · a957f6e · a957f6e
1 parent c4d0134
commit a957f6e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/module_mrfa/raft.py b/module_mrfa/raft.py
@@ -3,7 +3,7 @@
 import torch
 import torch.nn.functional as F
 from einops import rearrange
-from timm.models.layers.weight_init import trunc_normal_
+from timm.models.layers import trunc_normal_
 from torch import nn
 
 from .generator import OcclusionAwareGenerator

diff --git a/module_mrfa/transformer/tokenpose_base.py b/module_mrfa/transformer/tokenpose_base.py
@@ -3,7 +3,7 @@
 import torch
 import torch.nn.functional as F
 from einops import rearrange, repeat
-from timm.models.layers.weight_init import trunc_normal_
+from timm.models.layers import trunc_normal_
 from torch import nn
 
 from ..util import make_coordinate_grid

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-firstordermm"
 description = "ComfyUI-native nodes to run First Order Motion Model for Image Animation and its non-diffusion-based successors. [a/https://github.com/AliaksandrSiarohin/first-order-model](https://github.com/AliaksandrSiarohin/first-order-model)"
-version = "1.0.6"
+version = "1.0.7"
 license = { file = "LICENSE" }
 dependencies = ["numpy", "torch", "scipy", "pyyaml", "matplotlib", "einops", "timm"]