small tweaks

Author: Ubuntu

src/pruna/algorithms/zipar/utils.py

@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 from __future__ import annotations
 
 import copy
@@ -20,11 +19,15 @@
 import numpy as np
 import torch
 from transformers.generation.configuration_utils import GenerationMode
-from transformers.generation.logits_process import ClassifierFreeGuidanceLogitsProcessor, LogitsProcessorList
+from transformers.generation.logits_process import (
+    ClassifierFreeGuidanceLogitsProcessor,
+    LogitsProcessorList,
+)
 from transformers.generation.utils import GenerateDecoderOnlyOutput
 
 from pruna.logging.logger import pruna_logger
 
+
 class ZipARGenerationGrid:
     """
     Configuration and state tracking for ZipAR fast decoding.