additional styling

Author: yousef-rafat

comfy/ldm/higgsv2/preprocess.py

@@ -366,13 +366,13 @@ def convert_nan_to_none(obj):
                     audio_contents.append(content)
                     if role == "user" or role == "system":
                         text_tokens = tokenizer.encode(
-                            f"<|audio_bos|><|AUDIO|><|audio_eos|>",
+                            "<|audio_bos|><|AUDIO|><|audio_eos|>",
                             add_special_tokens=False,
                         )
                         input_tokens.extend(text_tokens)
                     elif role == "assistant":
                         text_tokens = tokenizer.encode(
-                            f"<|audio_out_bos|><|AUDIO_OUT|><|audio_eos|>",
+                            "<|audio_out_bos|><|AUDIO_OUT|><|audio_eos|>",
                             add_special_tokens=False,
                         )
                         input_tokens.extend(text_tokens)
@@ -587,7 +587,7 @@ def __call__(self, batch: List[ChatMLDatasetSample]):
             # I tried to remove the for-loop in original implementation
             # but to do batching with padding caused problem so I turned it into a list compre.
             lengths = [seg.shape[1] for seg in audio_in_ids_l]
-            aug_lengths = [l + 2 for l in lengths]
+            aug_lengths = [length + 2 for length in lengths]
             audio_in_ids_start = torch.cumsum(
                 torch.tensor([0] + aug_lengths[:-1], dtype=torch.long), dim=0
             )

comfy/ldm/higgsv2/tokenizer.py

@@ -1,4 +1,3 @@
-import os
 import math
 import torch
 import torch.nn as nn