kaushalya commited on
Commit
9cd84ba
1 Parent(s): b0632fa

Fix loading issue (#2)

Browse files

- Fix loading issue (a97b84b716321cd3c2c704ba9eea6e1e0693a584)

Files changed (1) hide show
  1. medclip/modeling_hybrid_clip.py +1 -0
medclip/modeling_hybrid_clip.py CHANGED
@@ -32,6 +32,7 @@ logger = logging.get_logger(__name__)
32
  class FlaxHybridCLIPModule(nn.Module):
33
  config: HybridCLIPConfig
34
  dtype: jnp.dtype = jnp.float32
 
35
 
36
  def setup(self):
37
  text_config = self.config.text_config
 
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