wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 13:50:15 +08:00
Code Issues Actions 15 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 import torch.nn as nn
from enum import Enum from enum import Enum
from dataclasses import dataclass from dataclasses import dataclass
from collections import defaultdict, OrderedDict from collections import OrderedDict
from typing import Optional, Tuple, Union, List from typing import Optional, Tuple, Union, List
class GenerationMode(Enum): class GenerationMode(Enum):
@ -1272,7 +1272,7 @@ class HiggsAudioModel(nn.Module):
for past_key_value in past_key_values: for past_key_value in past_key_values:
kv_cache_length = past_key_value.get_max_cache_shape() kv_cache_length = past_key_value.get_max_cache_shape()
for is_decoding_audio_token in [True, False]: for is_decoding_audio_token in [True, False]:
runner = CUDAGraphRunner(self._forward_core) runner = None#CUDAGraphRunner(self._forward_core)
batch_size = 1 batch_size = 1
hidden_dim = self.config["hidden_size"] hidden_dim = self.config["hidden_size"]