Jelajahi Sumber

feat:裁减kv_cache

zhaohaipeng 1 bulan lalu
induk
melakukan
77731d147e
1 mengubah file dengan 2 tambahan dan 0 penghapusan
  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