diff --git a/comfy/text_encoders/llama.py b/comfy/text_encoders/llama.py index f3a590dce..68ac1e804 100644 --- a/comfy/text_encoders/llama.py +++ b/comfy/text_encoders/llama.py @@ -1,7 +1,7 @@ import torch import torch.nn as nn from dataclasses import dataclass -from typing import Optional, Any, List, Tuple +from typing import Optional, Any, Tuple import math from comfy.ldm.modules.attention import optimized_attention_for_device