wedyanessam commited on
Commit
9afadf4
·
verified ·
1 Parent(s): 3e3a49a

Update FantasyTalking/model.py

Browse files
Files changed (1) hide show
  1. FantasyTalking/model.py +1 -1
FantasyTalking/model.py CHANGED
@@ -5,7 +5,7 @@ import torch.nn as nn
5
  import torch.nn.functional as F
6
  from safetensors import safe_open
7
 
8
- from Diffsynth.models.wan_video_dit import WanModel, flash_attention, attention
9
 
10
 
11
  class AudioProjModel(nn.Module):
 
5
  import torch.nn.functional as F
6
  from safetensors import safe_open
7
 
8
+ from FantasyTalking.Diffsynth.models.wan_video_dit import WanModel, flash_attention, attention
9
 
10
 
11
  class AudioProjModel(nn.Module):