potsawee commited on
Commit
e72ddf8
·
verified ·
1 Parent(s): 96a4150

Upload modeling_text_sync_mimi.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_text_sync_mimi.py +14 -6
modeling_text_sync_mimi.py CHANGED
@@ -4,12 +4,20 @@ import torch
4
  import torch.nn as nn
5
  from typing import Optional, Dict, List, Union
6
 
7
- from configuration_mimi import MimiConfig
8
- from modeling_mimi_clean import MimiPreTrainedModel, MimiModel
9
- from modeling_backbone_components import (
10
- CrossAttentionTransformer,
11
- CausalAttentionTransformer
12
- )
 
 
 
 
 
 
 
 
13
 
14
 
15
  class TextSyncMimi(MimiPreTrainedModel):
 
4
  import torch.nn as nn
5
  from typing import Optional, Dict, List, Union
6
 
7
+ try:
8
+ from .configuration_mimi import MimiConfig
9
+ from .modeling_mimi_clean import MimiPreTrainedModel, MimiModel
10
+ from .modeling_backbone_components import (
11
+ CrossAttentionTransformer,
12
+ CausalAttentionTransformer
13
+ )
14
+ except ImportError:
15
+ from configuration_mimi import MimiConfig
16
+ from modeling_mimi_clean import MimiPreTrainedModel, MimiModel
17
+ from modeling_backbone_components import (
18
+ CrossAttentionTransformer,
19
+ CausalAttentionTransformer
20
+ )
21
 
22
 
23
  class TextSyncMimi(MimiPreTrainedModel):