Vision-CAIR
commited on
Commit
•
2ec8ca7
1
Parent(s):
d119b40
Upload folder using huggingface_hub
Browse files- blip_processors.py +1 -0
- registry.py +0 -1
blip_processors.py
CHANGED
@@ -13,6 +13,7 @@ from .randaugment import RandomAugment
|
|
13 |
from omegaconf import OmegaConf
|
14 |
from torchvision import transforms
|
15 |
from torchvision.transforms.functional import InterpolationMode
|
|
|
16 |
|
17 |
|
18 |
class BlipImageBaseProcessor(BaseProcessor):
|
|
|
13 |
from omegaconf import OmegaConf
|
14 |
from torchvision import transforms
|
15 |
from torchvision.transforms.functional import InterpolationMode
|
16 |
+
print("blip_processors.py is imported")
|
17 |
|
18 |
|
19 |
class BlipImageBaseProcessor(BaseProcessor):
|
registry.py
CHANGED
@@ -243,7 +243,6 @@ class Registry:
|
|
243 |
|
244 |
@classmethod
|
245 |
def get_processor_class(cls, name):
|
246 |
-
print(cls.mapping["processor_name_mapping"])
|
247 |
return cls.mapping["processor_name_mapping"].get(name, None)
|
248 |
|
249 |
@classmethod
|
|
|
243 |
|
244 |
@classmethod
|
245 |
def get_processor_class(cls, name):
|
|
|
246 |
return cls.mapping["processor_name_mapping"].get(name, None)
|
247 |
|
248 |
@classmethod
|