kaushalya commited on
Commit
d445e6e
1 Parent(s): a269b46

Fix model loading issue

Browse files
Files changed (1) hide show
  1. medclip/modeling_hybrid_clip.py +1 -1
medclip/modeling_hybrid_clip.py CHANGED
@@ -32,7 +32,7 @@ logger = logging.get_logger(__name__)
32
  class FlaxHybridCLIPModule(nn.Module):
33
  config: HybridCLIPConfig
34
  dtype: jnp.dtype = jnp.float32
35
- _do_init: bool = False
36
 
37
  def setup(self):
38
  text_config = self.config.text_config
 
32
  class FlaxHybridCLIPModule(nn.Module):
33
  config: HybridCLIPConfig
34
  dtype: jnp.dtype = jnp.float32
35
+ _do_init: bool = True
36
 
37
  def setup(self):
38
  text_config = self.config.text_config