Spaces:

mehdi999
/

pardi-speech

Running on Zero

mehdi999 commited on 25 days ago

Commit

b8aa630

1 Parent(s): b650fff

fix: replace zcodec import with relative (TransformerBlock)

Files changed (1) hide show

codec/models/patchvae/modules.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch.nn.functional as F
 from torch import nn
 from vector_quantize_pytorch import FSQ
-from zcodec.models.components.transformer import TransformerBlock
 class AdaLayerNormScale(nn.Module):

 from torch import nn
 from vector_quantize_pytorch import FSQ
+from ..components.transformer import TransformerBlock
 class AdaLayerNormScale(nn.Module):