Add input metadata

2023-02-22 19:01:20 +00:00
parent 608f74ffe5
commit 7b6844e590
2 changed files with 28 additions and 2 deletions
--- a/cacheflow/models/init.py
+++ b/cacheflow/models/init.py
@@ -1,7 +1,8 @@
-from cacheflow.worker.models.model_utils import get_model
+from cacheflow.models.input_metadata import InputMetadata
+from cacheflow.models.model_utils import get_model


 __all__ = [
    'get_model',
-    
+    'InputMetadata',
 ]