review: remove commented import

huggingface · Oct 11, 2024 · fa4b706 · fa4b706
1 parent 3b0d2e0
commit fa4b706
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/...ion-inference/server/text_generation_server/jetstream_pt_support/models/gemma_model_hf.py b/...ion-inference/server/text_generation_server/jetstream_pt_support/models/gemma_model_hf.py
@@ -1,8 +1,6 @@
 
 from jetstream_pt.third_party.gemma import config as gemma_config
 from jetstream_pt.third_party.gemma.model import GemmaModel
-
-#.model_exportable import Transformer, model_args
 from transformers import GemmaConfig, GenerationConfig, GenerationMixin
 
 

diff --git a/...n-inference/server/text_generation_server/jetstream_pt_support/models/mixtral_model_hf.py b/...n-inference/server/text_generation_server/jetstream_pt_support/models/mixtral_model_hf.py
@@ -1,9 +1,7 @@
 
 from jetstream_pt.third_party.mixtral import config as mixtral_config
 from jetstream_pt.third_party.mixtral.model import Transformer
-
-#.model_exportable import Transformer, model_args
-from transformers import MixtralConfig, GenerationConfig, GenerationMixin
+from transformers import GenerationConfig, GenerationMixin, MixtralConfig
 
 
 class MixtralModelHf(Transformer, GenerationMixin):