chore: format with ruff

Author: tsdocode

dia/layers.py

@@ -430,15 +430,12 @@ def patch_fused_qkv(self):
 
         self.qkv = FusedQKV(
             self.kv_embed_dim,
-            (
-                self.num_query_heads * self.head_dim
-                + 2 * (self.num_kv_heads * self.head_dim)
-            ),
+            (self.num_query_heads * self.head_dim + 2 * (self.num_kv_heads * self.head_dim)),
             bias=False,
             num_q_heads=self.num_query_heads,
             q_head_dim=self.head_dim,
             num_kv_heads=self.num_kv_heads,
-            kv_head_dim=self.head_dim
+            kv_head_dim=self.head_dim,
         )
         self.qkv.linear.weight.data = torch.cat([q_proj_weight, k_proj_weight, v_proj_weight], dim=0)
 

dia/model.py

@@ -372,7 +372,12 @@ def _prepare_generation(
             encoder_out, enc_state.positions, enc_state.padding_mask
         )
         dec_state = DecoderInferenceState.new(
-            self.config, enc_state, encoder_out, dec_cross_attn_cache, self.compute_dtype, max_generation_length=max_tokens
+            self.config,
+            enc_state,
+            encoder_out,
+            dec_cross_attn_cache,
+            self.compute_dtype,
+            max_generation_length=max_tokens,
         )
         prefill, prefill_steps = self._prepare_audio_prompt(audio_prompts)