From e12133613bac83945dca344b86317fdf62f34f9c Mon Sep 17 00:00:00 2001 From: Casper Date: Thu, 7 Dec 2023 21:58:29 +0100 Subject: [PATCH] Remove another unused import --- src/axolotl/monkeypatch/llama_attn_hijack_flash.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/axolotl/monkeypatch/llama_attn_hijack_flash.py b/src/axolotl/monkeypatch/llama_attn_hijack_flash.py index 108e30bab..ea9ce75ca 100644 --- a/src/axolotl/monkeypatch/llama_attn_hijack_flash.py +++ b/src/axolotl/monkeypatch/llama_attn_hijack_flash.py @@ -6,7 +6,6 @@ import logging from typing import List, Optional, Tuple, Union import torch -import torch.nn.functional as F import transformers from transformers.modeling_outputs import BaseModelOutputWithPast from transformers.models.llama.modeling_llama import LlamaAttention