From 7c6b9a096353401ab6b0d92cecb30ef61089d17d Mon Sep 17 00:00:00 2001 From: fxmarty <9808326+fxmarty@users.noreply.github.com> Date: Fri, 17 May 2024 08:50:20 +0000 Subject: [PATCH] remove unnecessary imports --- server/text_generation_server/models/flash_gpt2.py | 1 - server/text_generation_server/models/flash_mistral.py | 1 - server/text_generation_server/models/flash_neox.py | 1 - server/text_generation_server/models/flash_phi.py | 1 - server/text_generation_server/models/flash_qwen2.py | 1 - 5 files changed, 5 deletions(-) diff --git a/server/text_generation_server/models/flash_gpt2.py b/server/text_generation_server/models/flash_gpt2.py index 2511148d..65bb9920 100644 --- a/server/text_generation_server/models/flash_gpt2.py +++ b/server/text_generation_server/models/flash_gpt2.py @@ -15,7 +15,6 @@ from text_generation_server.utils import ( weight_files, Weights, ) -from text_generation_server.models import CausalLM from text_generation_server.utils.import_utils import SYSTEM tracer = trace.get_tracer(__name__) diff --git a/server/text_generation_server/models/flash_mistral.py b/server/text_generation_server/models/flash_mistral.py index 295fcc41..5f0d20f5 100644 --- a/server/text_generation_server/models/flash_mistral.py +++ b/server/text_generation_server/models/flash_mistral.py @@ -27,7 +27,6 @@ from text_generation_server.utils import ( HeterogeneousNextTokenChooser, StoppingCriteria, ) -from text_generation_server.models import CausalLM tracer = trace.get_tracer(__name__) diff --git a/server/text_generation_server/models/flash_neox.py b/server/text_generation_server/models/flash_neox.py index fe1ebbaf..9242a124 100644 --- a/server/text_generation_server/models/flash_neox.py +++ b/server/text_generation_server/models/flash_neox.py @@ -15,7 +15,6 @@ from text_generation_server.utils import ( Weights, ) from text_generation_server.utils.import_utils import SYSTEM -from text_generation_server.models import CausalLM tracer = trace.get_tracer(__name__) diff --git a/server/text_generation_server/models/flash_phi.py b/server/text_generation_server/models/flash_phi.py index 1934254e..92f0a2b5 100644 --- a/server/text_generation_server/models/flash_phi.py +++ b/server/text_generation_server/models/flash_phi.py @@ -15,7 +15,6 @@ from text_generation_server.utils import ( weight_files, Weights, ) -from text_generation_server.models import CausalLM tracer = trace.get_tracer(__name__) diff --git a/server/text_generation_server/models/flash_qwen2.py b/server/text_generation_server/models/flash_qwen2.py index 511e33e0..5ee8ffe6 100644 --- a/server/text_generation_server/models/flash_qwen2.py +++ b/server/text_generation_server/models/flash_qwen2.py @@ -20,7 +20,6 @@ from text_generation_server.utils import ( weight_files, Weights, ) -from text_generation_server.models import CausalLM tracer = trace.get_tracer(__name__)