mehdi999 commited on
Commit
a0c7117
·
1 Parent(s): 8d935ec
Files changed (1) hide show
  1. codec/models/patchvae/modules.py +1 -1
codec/models/patchvae/modules.py CHANGED
@@ -6,7 +6,7 @@ import torch.nn.functional as F
6
  from torch import nn
7
  from vector_quantize_pytorch import FSQ
8
 
9
- from zcodec.models.components.transformer import TransformerBlock
10
 
11
 
12
  class AdaLayerNormScale(nn.Module):
 
6
  from torch import nn
7
  from vector_quantize_pytorch import FSQ
8
 
9
+ from ..components.transformer import TransformerBlock
10
 
11
 
12
  class AdaLayerNormScale(nn.Module):