diff --git a/comfy/ldm/higgsv2/model.py b/comfy/ldm/higgsv2/model.py
index 7d08edade..a8a811bab 100644
--- a/comfy/ldm/higgsv2/model.py
+++ b/comfy/ldm/higgsv2/model.py
@@ -14,7 +14,7 @@ import torch
 import torch.nn as nn
 from enum import Enum
 from dataclasses import dataclass
-from collections import defaultdict, OrderedDict
+from collections import OrderedDict
 from typing import Optional, Tuple, Union, List
 
 class GenerationMode(Enum):
@@ -1272,7 +1272,7 @@ class HiggsAudioModel(nn.Module):
         for past_key_value in past_key_values:
             kv_cache_length = past_key_value.get_max_cache_shape()
             for is_decoding_audio_token in [True, False]:
-                runner = CUDAGraphRunner(self._forward_core)
+                runner = None#CUDAGraphRunner(self._forward_core)
 
                 batch_size = 1
                 hidden_dim = self.config["hidden_size"]