Fix loading issue

#2
by kaushalya - opened
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