Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ sys.path.insert(0, metalatte_path)
|
|
7 |
|
8 |
# Import the custom configuration and model
|
9 |
from configuration import MetaLATTEConfig
|
10 |
-
from
|
11 |
AutoConfig.register("metalatte", MetaLATTEConfig)
|
12 |
AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
|
13 |
|
|
|
7 |
|
8 |
# Import the custom configuration and model
|
9 |
from configuration import MetaLATTEConfig
|
10 |
+
from modeling_metalatte import MetaLATTEModel
|
11 |
AutoConfig.register("metalatte", MetaLATTEConfig)
|
12 |
AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
|
13 |
|