Lint and fix undefined names (1/N) (#6028)

This commit is contained in:
Chenlei Hu
2024-12-12 15:55:26 -08:00
committed by GitHub
parent 60749f345d
commit 2cddbf0821
5 changed files with 15 additions and 9 deletions

View File

@@ -1,3 +1,4 @@
import math
import torch
from torch import nn
from .ldm.modules.attention import CrossAttention