nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

exclude enhance-a-video from compilation

Browse Source
This commit is contained in:
kijai
2025-03-09 23:52:18 +02:00
parent 07848ab0b5
commit 1fe746bb9d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance_a_video/enhance.py
+2
View File
@@ -2,6 +2,7 @@ import torch
from einops import rearrange from einops import rearrange
from .globals import get_enhance_weight, get_num_frames from .globals import get_enhance_weight, get_num_frames
@torch.compiler.disable()
def get_feta_scores(query, key): def get_feta_scores(query, key):
img_q, img_k = query, key img_q, img_k = query, key
@@ -26,6 +27,7 @@ def get_feta_scores(query, key):
return feta_score(query_image, key_image, C, num_frames) return feta_score(query_image, key_image, C, num_frames)
@torch.compiler.disable()
def feta_score(query_image, key_image, head_dim, num_frames): def feta_score(query_image, key_image, head_dim, num_frames):
scale = head_dim**-0.5 scale = head_dim**-0.5
query_image = query_image * scale query_image = query_image * scale