choiking10 / koclip Goto Github PK

TypeError                                 Traceback (most recent call last)
Cell In[1], line 7
      3 from PIL import Image
      5 from koclip import load_koclip
----> 7 model, processor = load_koclip("koclip-base")

File ~/project/koclip/koclip/utils.py:9, in load_koclip(model_name)
      7 def load_koclip(model_name):
      8     assert model_name in {"koclip-base", "koclip-large"}
----> 9     model = FlaxHybridCLIP.from_pretrained(f"koclip/{model_name}")
     10     processor = FlaxHybridCLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
     11     processor.tokenizer = AutoTokenizer.from_pretrained("klue/roberta-large")

File ~/project/koclip/venv/lib/python3.9/site-packages/transformers/modeling_flax_utils.py:808, in FlaxPreTrainedModel.from_pretrained(cls, pretrained_model_name_or_path, dtype, *model_args, **kwargs)
    792     resolved_archive_file, _ = get_checkpoint_shard_files(
    793         pretrained_model_name_or_path,
    794         resolved_archive_file,
   (...)
    804         _commit_hash=commit_hash,
    805     )
    807 # init random models
--> 808 model = cls(config, *model_args, _do_init=_do_init, **model_kwargs)
    810 if from_pt:
    811     state = load_pytorch_checkpoint_in_flax_state_dict(model, resolved_archive_file, is_sharded)

File ~/project/koclip/koclip/model.py:161, in FlaxHybridCLIP.__init__(self, config, input_shape, seed, dtype, **kwargs)
    150 if input_shape is None:
    151     input_shape = (
    152         (1, 1),
    153         (
   (...)
    158         ),
    159     )
--> 161 module = self.module_class(config=config, dtype=dtype, **kwargs)
    162 super().__init__(
    163     config, module, input_shape=input_shape, seed=seed, dtype=dtype
    164 )

TypeError: __init__() got an unexpected keyword argument '_do_init'