Spaces:
Runtime error
Runtime error
Merge branch 'opencode/playful-orchid'
Browse files- shared/model_manager.py +1 -1
shared/model_manager.py
CHANGED
|
@@ -142,7 +142,7 @@ class ModelManager:
|
|
| 142 |
PositionalEncoding as MuseTalkPositionalEncoding,
|
| 143 |
)
|
| 144 |
|
| 145 |
-
pe =
|
| 146 |
|
| 147 |
self._musetalk_unet = unet
|
| 148 |
self._musetalk_pe = pe
|
|
|
|
| 142 |
PositionalEncoding as MuseTalkPositionalEncoding,
|
| 143 |
)
|
| 144 |
|
| 145 |
+
pe = MuseTalkPositionalEncoding(d_model=384)
|
| 146 |
|
| 147 |
self._musetalk_unet = unet
|
| 148 |
self._musetalk_pe = pe
|