Spaces:

flax-community
/

clip-reply-demo

Runtime error

ceyda commited on May 16, 2022

Commit

e438e9d

•

1 Parent(s): 5ffe5df

Update model/model.py

Files changed (1) hide show

model/model.py CHANGED Viewed

@@ -157,7 +157,7 @@ class FlaxHybridCLIP(FlaxPreTrainedModel):
                     3,
                 ),
             )
         module = self.module_class(config=config, dtype=dtype, **kwargs)
         super().__init__(
             config, module, input_shape=input_shape, seed=seed, dtype=dtype

                     3,
                 ),
             )
+        kwargs.pop('_do_init', None) # temp fix possibly related: https://github.com/huggingface/transformers/issues/15766
         module = self.module_class(config=config, dtype=dtype, **kwargs)
         super().__init__(
             config, module, input_shape=input_shape, seed=seed, dtype=dtype