From dc1a0f41dbd6ae4699cd0f667770fa27fb1dfd79 Mon Sep 17 00:00:00 2001
From: Aleksandr Borzunov <hxrussia@gmail.com>
Date: Wed, 24 Jul 2024 12:45:47 +0000
Subject: [PATCH] Remove unused imports

---
 src/petals/models/bloom/block.py   | 2 +-
 src/petals/models/llama/block.py   | 1 -
 src/petals/models/mixtral/block.py | 3 +--
 3 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/petals/models/bloom/block.py b/src/petals/models/bloom/block.py
index 439b9ca10..01a74b216 100644
--- a/src/petals/models/bloom/block.py
+++ b/src/petals/models/bloom/block.py
@@ -7,7 +7,7 @@
 
 import torch
 from transformers.modeling_attn_mask_utils import _prepare_4d_causal_attention_mask
-from transformers.models.bloom.modeling_bloom import BloomBlock, BloomModel, build_alibi_tensor
+from transformers.models.bloom.modeling_bloom import BloomBlock, build_alibi_tensor
 
 from petals.utils.misc import is_dummy
 
diff --git a/src/petals/models/llama/block.py b/src/petals/models/llama/block.py
index 1c89ad3f5..4ff9d3f69 100644
--- a/src/petals/models/llama/block.py
+++ b/src/petals/models/llama/block.py
@@ -15,7 +15,6 @@
     LlamaConfig,
     LlamaDecoderLayer,
     LlamaMLP,
-    LlamaModel,
     LlamaRMSNorm,
     repeat_kv,
     rotate_half,
diff --git a/src/petals/models/mixtral/block.py b/src/petals/models/mixtral/block.py
index 7a2bd9fe3..58acd1444 100644
--- a/src/petals/models/mixtral/block.py
+++ b/src/petals/models/mixtral/block.py
@@ -1,4 +1,3 @@
-import json
 from typing import Optional, Tuple
 
 import torch
@@ -8,7 +7,7 @@
     _prepare_4d_causal_attention_mask,
     _prepare_4d_causal_attention_mask_for_sdpa,
 )
-from transformers.models.mixtral.modeling_mixtral import MixtralDecoderLayer, MixtralModel
+from transformers.models.mixtral.modeling_mixtral import MixtralDecoderLayer
 
 
 class WrappedMixtralBlock(MixtralDecoderLayer):