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 dataclasses import dataclass
 from typing import List, Optional, Union
 from typing import List, Optional, Union
 
 
+import numpy as np
 import torch
 import torch
+from audiotools import AudioSignal
 from audiotools.ml import BaseModel
 from audiotools.ml import BaseModel
 from dac.model.base import CodecMixin
 from dac.model.base import CodecMixin
 from dac.nn.layers import Snake1d, WNConv1d, WNConvTranspose1d
 from dac.nn.layers import Snake1d, WNConv1d, WNConvTranspose1d