فهرست منبع

feat:裁减kv_cache

zhaohaipeng 1 ماه پیش
والد
کامیت
77731d147e
1فایلهای تغییر یافته به همراه2 افزوده شده و 0 حذف شده
  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