Upload modeling_text_sync_mimi.py with huggingface_hub
Browse files- 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 |
-
|
| 8 |
-
from
|
| 9 |
-
from
|
| 10 |
-
|
| 11 |
-
|
| 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):
|