Vision-CAIR commited on
Commit
2ec8ca7
1 Parent(s): d119b40

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. blip_processors.py +1 -0
  2. 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