diff --git a/comfy/ldm/cosmos/predict2.py b/comfy/ldm/cosmos/predict2.py
index 89a80ca4f..3b91b3f6e 100644
--- a/comfy/ldm/cosmos/predict2.py
+++ b/comfy/ldm/cosmos/predict2.py
@@ -5,9 +5,8 @@ from torch import nn
 from einops import rearrange
 from einops.layers.torch import Rearrange
 import logging
-from typing import Any, Callable, List, Optional, Tuple
+from typing import Callable, Optional, Tuple
 import math
-import numpy as np
 
 from .position_embedding import VideoRopePosition3DEmb, LearnablePosEmbAxis
 from torchvision import transforms