Post-merge fixes

bigscience-workshop · Oct 20, 2023 · d87c23b · d87c23b
1 parent 86a1aef
commit d87c23b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/src/petals/models/llama/block.py b/src/petals/models/llama/block.py
@@ -4,7 +4,6 @@
 See commit history for authorship.
 """
 import math
-import math
 from typing import Optional, Tuple
 
 import torch

diff --git a/tests/test_optimized_layers.py b/tests/test_optimized_layers.py
@@ -4,7 +4,6 @@
 import torch
 from transformers.models.falcon.modeling_falcon import FalconDecoderLayer, FalconModel, build_alibi_tensor
 from transformers.models.llama.modeling_llama import LlamaDecoderLayer, LlamaModel
-from transformers.models.llama.modeling_llama import LlamaDecoderLayer, LlamaModel
 
 from petals.utils.auto_config import AutoDistributedConfig
 from petals.utils.convert_block import QuantType, convert_block