fix demo issues

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+GenAU/src/modules/clap/open_clip/bpe_simple_vocab_16e6.txt.gz filter=lfs diff=lfs merge=lfs -text

GenAU/src/models/genau_ddpm.py

@@ -25,7 +25,7 @@ from pytorch_lightning.utilities.rank_zero import rank_zero_only
 # requires latest versin of torch
 from torch.utils.flop_counter import FlopCounterMode
 
-from src.modules.conditional.conditional_models import *
+from src.modules.conditional.conditional_models import CLAPAudioEmbeddingClassifierFreev2
 from src.utilities.model.model_util import (
     exists,
     default,

GenAU/src/modules/conditional/conditional_models.py

@@ -26,8 +26,12 @@ from src.modules.audiomae.sequence_gen.sequence_input import (
 from src.tools.download_manager import get_checkpoint_path
 import numpy as np
 from src.modules.audiomae.sequence_gen.model import Prenet
-import laion_clap
 
+try:
+    import laion_clap
+except ImportError:
+    print("[Warnning] laion_clap not found. Please install it from")
+    pass
 
 """
 The model forward function can return three types of data:

requirements.txt

@@ -44,5 +44,7 @@ huggingface_hub==0.28.1
 taming-transformers-rom1504==0.0.6
 warmup_scheduler==0.3
 ruamel.yaml==0.18.6
+flash-attn==2.3.0
+-e pip uninstall -y apex
 # git+https://github.com/LAION-AI/CLAP.git
 # sentence_transformers==2.7.0