wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-15 16:50:57 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Yousef Rafat 2025-10-24 18:22:02 +03:00
parent faa866733d
commit 01b32521c5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/higgsv2/model.py
View File

@ -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"]