Muennighoff
commited on
Commit
•
3f4d90d
1
Parent(s):
996201e
Fix import
Browse files- modeling_molmo.py +1 -1
modeling_molmo.py
CHANGED
@@ -13,7 +13,7 @@ from transformers.modeling_outputs import CausalLMOutputWithPast, ModelOutput
|
|
13 |
from transformers.models.auto import AutoModelForCausalLM
|
14 |
from torch import nn
|
15 |
|
16 |
-
from
|
17 |
from torch.nn import functional as F
|
18 |
|
19 |
|
|
|
13 |
from transformers.models.auto import AutoModelForCausalLM
|
14 |
from torch import nn
|
15 |
|
16 |
+
from .config_molmo import MolmoConfig
|
17 |
from torch.nn import functional as F
|
18 |
|
19 |
|