Spaces:
Running
Running
Fix loading issue (#2)
Browse files- Fix loading issue (a97b84b716321cd3c2c704ba9eea6e1e0693a584)
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
|