Vision-CAIR commited on
Commit
92fb897
·
verified ·
1 Parent(s): 7595a0d

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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):