lint

huggingface · Feb 1, 2025 · 734c605 · 734c605
1 parent 1c8a02c
commit 734c605
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/exporters/neuron/model_configs/decoder_configs.py b/optimum/exporters/neuron/model_configs/decoder_configs.py
@@ -14,7 +14,6 @@
 # limitations under the License.
 """Neuron export configurations for models using transformers_neuronx."""
 
-
 from optimum.exporters.tasks import TasksManager
 
 from ....neuron.models.granite.model import GraniteForSampling
@@ -72,6 +71,7 @@ class GraniteNeuronConfig(TextNeuronDecoderConfig):
     NEURONX_CLASS = GraniteForSampling
     CONTINUOUS_BATCHING = True
 
+
 @register_in_tasks_manager("phi4", "text-generation")
 class Phi4NeuronConfig(TextNeuronDecoderConfig):
     NEURONX_CLASS = Phi4ForSampling