trying to cast

vui/inference.py

@@ -5,6 +5,7 @@ import inflect
 import torch
 import torch.nn.functional as F
 import torchaudio
+from torchaudio.transforms import Resample
 from torch import Tensor
 from torch.nn.attention import SDPBackend, sdpa_kernel
 
@@ -12,6 +13,7 @@ from vui.model import Vui
 from vui.sampling import multinomial, sample_top_k, sample_top_p, sample_top_p_top_k
 from vui.vad import detect_voice_activity as vad
 
+resample = Resample(22050, 16000).cuda()
 
 def ensure_spaces_around_tags(text: str):
     # Add space before '[' if not preceded by space, '<', or '['
@@ -328,7 +330,8 @@ def render(
         )
         codes = codes[..., :-10]
         audio = self.codec.from_indices(codes)
-        paudio = torchaudio.functional.resample(audio[0], 22050, 16000)
+        
+        paudio = resample(audio[0])
         results = vad(paudio)
 
         if len(results):