Spaces:

FunAudioLLM
/

PrismAudio

Running on Zero

liuhuadai commited on about 14 hours ago

Commit

27f9239

verified ·

1 Parent(s): 9ebfa72

Update data_utils/v2a_utils/feature_utils_288.py

Files changed (1) hide show

data_utils/v2a_utils/feature_utils_288.py CHANGED Viewed

@@ -93,7 +93,7 @@ class FeaturesUtils(nn.Module):
     def _init_jax(self):
-        if hasattr(self, "flax_model"):
             return  # already init
         backend = jax.default_backend()
         if backend != 'gpu':

     def _init_jax(self):
+        if hasattr(self, "flax_model") and hasattr(self, "text_tokenizer"):
             return  # already init
         backend = jax.default_backend()
         if backend != 'gpu':