Update modeling_flamingo.py
Browse files- modeling_flamingo.py +2 -2
modeling_flamingo.py
CHANGED
@@ -20,8 +20,8 @@ try:
|
|
20 |
except:
|
21 |
_prepare_4d_causal_attention_mask = None
|
22 |
|
23 |
-
from utils import exists, freeze_all_layers_, unfreeze_all_layers_
|
24 |
-
from flamingo_pytorch import GatedCrossAttentionBlock, PerceiverResampler
|
25 |
from .configuration_flamingo import FlamingoConfig
|
26 |
|
27 |
|
|
|
20 |
except:
|
21 |
_prepare_4d_causal_attention_mask = None
|
22 |
|
23 |
+
from .utils import exists, freeze_all_layers_, unfreeze_all_layers_
|
24 |
+
from .flamingo_pytorch import GatedCrossAttentionBlock, PerceiverResampler
|
25 |
from .configuration_flamingo import FlamingoConfig
|
26 |
|
27 |
|