added elastic model

app.py

@@ -8,41 +8,41 @@ import subprocess
 import sys
 import os
 
-def setup_flash_attention():
-    """One-time setup for flash-attention with special flags"""
-    # Check if flash-attn is already installed
-    try:
-        import flash_attn
-        print("flash-attn already installed")
-        return
-    except ImportError:
-        pass
+# def setup_flash_attention():
+#     """One-time setup for flash-attention with special flags"""
+#     # Check if flash-attn is already installed
+#     try:
+#         import flash_attn
+#         print("flash-attn already installed")
+#         return
+#     except ImportError:
+#         pass
     
-    # Check if we've already tried to install it in this session
-    if os.path.exists("/tmp/flash_attn_installed"):
-        return
+#     # Check if we've already tried to install it in this session
+#     if os.path.exists("/tmp/flash_attn_installed"):
+#         return
         
-    try:
-        print("Installing flash-attn with --no-build-isolation...")
-        subprocess.run([
-            sys.executable, "-m", "pip", "install", 
-            "flash-attn==2.7.3", "--no-build-isolation"
-        ], check=True)
+#     try:
+#         print("Installing flash-attn with --no-build-isolation...")
+#         subprocess.run([
+#             sys.executable, "-m", "pip", "install", 
+#             "flash-attn==2.7.3", "--no-build-isolation"
+#         ], check=True)
         
-        # Uninstall apex if it exists
-        subprocess.run([
-            sys.executable, "-m", "pip", "uninstall", "apex", "-y"
-        ], check=False)  # Don't fail if apex isn't installed
+#         # Uninstall apex if it exists
+#         subprocess.run([
+#             sys.executable, "-m", "pip", "uninstall", "apex", "-y"
+#         ], check=False)  # Don't fail if apex isn't installed
         
-        # Mark as installed
-        with open("/tmp/flash_attn_installed", "w") as f:
-            f.write("installed")
+#         # Mark as installed
+#         with open("/tmp/flash_attn_installed", "w") as f:
+#             f.write("installed")
             
-        print("flash-attn installation completed")
+#         print("flash-attn installation completed")
         
-    except subprocess.CalledProcessError as e:
-        print(f"Warning: Failed to install flash-attn: {e}")
-        # Continue anyway - the model might work without it
+#     except subprocess.CalledProcessError as e:
+#         print(f"Warning: Failed to install flash-attn: {e}")
+#         # Continue anyway - the model might work without it
 
 # Run setup once when the module is imported
 # setup_flash_attention()
@@ -85,8 +85,7 @@ def generate_music(text_prompt, duration=10, temperature=1.0, top_k=250, top_p=0
                 cache_implementation="paged"
             )
         
-        # Convert to numpy array and prepare for output
-        audio_data = audio_values[0, 0].cpu().numpy()
+        audio_data = audio_values[0, 0].cpu().numpy().astype(np.float32)
         sample_rate = model.config.sample_rate
         
         # Normalize audio
@@ -95,7 +94,8 @@ def generate_music(text_prompt, duration=10, temperature=1.0, top_k=250, top_p=0
         return sample_rate, audio_data
         
     except Exception as e:
-        return None, f"Error generating music: {str(e)}"
+        print(f"Error: {str(e)}")
+        return None
 
 # Create Gradio interface
 with gr.Blocks(title="MusicGen Large - Music Generation") as demo: