Upload folder using huggingface_hub
Browse files- registry.py +0 -60
registry.py
CHANGED
@@ -18,66 +18,6 @@ class Registry:
|
|
18 |
"paths": {},
|
19 |
}
|
20 |
|
21 |
-
@classmethod
|
22 |
-
def register_builder(cls, name):
|
23 |
-
r"""Register a dataset builder to registry with key 'name'
|
24 |
-
|
25 |
-
Args:
|
26 |
-
name: Key with which the builder will be registered.
|
27 |
-
|
28 |
-
Usage:
|
29 |
-
|
30 |
-
from minigpt4.common.registry import registry
|
31 |
-
from minigpt4.datasets.base_dataset_builder import BaseDatasetBuilder
|
32 |
-
"""
|
33 |
-
|
34 |
-
def wrap(builder_cls):
|
35 |
-
from minigpt4.datasets.builders.base_dataset_builder import BaseDatasetBuilder
|
36 |
-
|
37 |
-
assert issubclass(
|
38 |
-
builder_cls, BaseDatasetBuilder
|
39 |
-
), "All builders must inherit BaseDatasetBuilder class, found {}".format(
|
40 |
-
builder_cls
|
41 |
-
)
|
42 |
-
if name in cls.mapping["builder_name_mapping"]:
|
43 |
-
raise KeyError(
|
44 |
-
"Name '{}' already registered for {}.".format(
|
45 |
-
name, cls.mapping["builder_name_mapping"][name]
|
46 |
-
)
|
47 |
-
)
|
48 |
-
cls.mapping["builder_name_mapping"][name] = builder_cls
|
49 |
-
return builder_cls
|
50 |
-
|
51 |
-
return wrap
|
52 |
-
|
53 |
-
@classmethod
|
54 |
-
def register_task(cls, name):
|
55 |
-
r"""Register a task to registry with key 'name'
|
56 |
-
|
57 |
-
Args:
|
58 |
-
name: Key with which the task will be registered.
|
59 |
-
|
60 |
-
Usage:
|
61 |
-
|
62 |
-
from minigpt4.common.registry import registry
|
63 |
-
"""
|
64 |
-
|
65 |
-
def wrap(task_cls):
|
66 |
-
from minigpt4.tasks.base_task import BaseTask
|
67 |
-
|
68 |
-
assert issubclass(
|
69 |
-
task_cls, BaseTask
|
70 |
-
), "All tasks must inherit BaseTask class"
|
71 |
-
if name in cls.mapping["task_name_mapping"]:
|
72 |
-
raise KeyError(
|
73 |
-
"Name '{}' already registered for {}.".format(
|
74 |
-
name, cls.mapping["task_name_mapping"][name]
|
75 |
-
)
|
76 |
-
)
|
77 |
-
cls.mapping["task_name_mapping"][name] = task_cls
|
78 |
-
return task_cls
|
79 |
-
|
80 |
-
return wrap
|
81 |
|
82 |
@classmethod
|
83 |
def register_model(cls, name):
|
|
|
18 |
"paths": {},
|
19 |
}
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
@classmethod
|
23 |
def register_model(cls, name):
|