Commit
•
4993d3c
1
Parent(s):
a71c06f
Update audioldm_train/modules/diffusionmodules/PixArt.py
Browse files
audioldm_train/modules/diffusionmodules/PixArt.py
CHANGED
|
@@ -149,7 +149,7 @@ class PixArtBlock(nn.Module):
|
|
| 149 |
|
| 150 |
return x
|
| 151 |
|
| 152 |
-
from audioldm_train.modules.diffusionmodules.attention import CrossAttention_1D
|
| 153 |
|
| 154 |
class PixArtBlock_Slow(nn.Module):
|
| 155 |
"""
|
|
|
|
| 149 |
|
| 150 |
return x
|
| 151 |
|
| 152 |
+
from qa_mdt.audioldm_train.modules.diffusionmodules.attention import CrossAttention_1D
|
| 153 |
|
| 154 |
class PixArtBlock_Slow(nn.Module):
|
| 155 |
"""
|