Fix ruff.

This commit is contained in:
comfyanonymous 2025-06-13 06:56:46 -04:00 committed by GitHub
parent 5355f8e793
commit eacded568b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5,9 +5,8 @@ from torch import nn
from einops import rearrange from einops import rearrange
from einops.layers.torch import Rearrange from einops.layers.torch import Rearrange
import logging import logging
from typing import Any, Callable, List, Optional, Tuple from typing import Callable, Optional, Tuple
import math import math
import numpy as np
from .position_embedding import VideoRopePosition3DEmb, LearnablePosEmbAxis from .position_embedding import VideoRopePosition3DEmb, LearnablePosEmbAxis
from torchvision import transforms from torchvision import transforms