Explorar o código

feat:裁减kv_cache

zhaohaipeng hai 1 mes
pai
achega
77731d147e
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      fish_speech/models/dac/modded_dac.py

+ 2 - 0
fish_speech/models/dac/modded_dac.py

@@ -3,7 +3,9 @@ import typing as tp
 from dataclasses import dataclass
 from typing import List, Optional, Union
 
+import numpy as np
 import torch
+from audiotools import AudioSignal
 from audiotools.ml import BaseModel
 from dac.model.base import CodecMixin
 from dac.nn.layers import Snake1d, WNConv1d, WNConvTranspose1d