apply suggestion

huggingface · Oct 23, 2024 · 519589a · 519589a
1 parent 0a3266d
commit 519589a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/neuron/modeling_seq2seq.py b/optimum/neuron/modeling_seq2seq.py
@@ -155,7 +155,7 @@ def load_model(
         decoder_path: Union[str, Path],
         tensor_parallel_size: int,
     ):
-        if not tensor_parallel_size > 1:
+        if tensor_parallel_size == 1:
             # Initialize Neuron Runtime before loading models
             runtime = torch.classes.neuron.Runtime()
             runtime.initialize()