diff --git a/modules/llama_attn_hijack.py b/modules/llama_attn_hijack.py index e4d1ceab..a7559a57 100644 --- a/modules/llama_attn_hijack.py +++ b/modules/llama_attn_hijack.py @@ -4,7 +4,7 @@ from typing import Optional, Tuple import torch import torch.nn as nn -import transformers.models.llama.modeling_llama + import modules.shared as shared from modules.logging_colors import logger