From 01b32521c528a22542c5ade9f4e1a50a0460f2af Mon Sep 17 00:00:00 2001 From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com> Date: Fri, 24 Oct 2025 18:22:02 +0300 Subject: [PATCH] . --- comfy/ldm/higgsv2/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"]