Spaces:
Running
Running
Fix model loading issue
Browse files
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 =
|
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
|