Procházet zdrojové kódy

feat:裁减kv_cache

zhaohaipeng před 1 měsícem
rodič
revize
77731d147e
1 změnil soubory, kde provedl 2 přidání a 0 odebrání
  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