Update app.py

app.py

@@ -77,9 +77,13 @@ def get_available_device():
 
 # print("FISH_AE_DTYPE:", FISH_AE_DTYPE)
 # print("IS_ZEROGPU:", IS_ZEROGPU)
-# device = get_available_device()
+device = get_available_device()
 # print(f"Using device: {device}")
 
+# Определяем, доступна ли компиляция (только для CUDA и не на Zero GPU)
+COMPILE_AVAILABLE = (device == 'cuda' and not IS_ZEROGPU)
+# print(f"Compile available: {COMPILE_AVAILABLE}")
+
 def _safe_members(tf, prefix):
     if not prefix.endswith('/'):
         prefix += '/'
@@ -180,7 +184,7 @@ def load_models():
         # print(f"Loading models to device: {device}")
         
         model = load_model_from_hf(dtype=MODEL_DTYPE, compile=False, token=HF_TOKEN, device=device)
-        fish_ae = load_fish_ae_from_hf(compile=(COMPILE_FISH_IF_NOT_ON_ZERO_GPU and not IS_ZEROGPU), dtype=FISH_AE_DTYPE, token=HF_TOKEN, device=device)
+        fish_ae = load_fish_ae_from_hf(compile=False, dtype=FISH_AE_DTYPE, token=HF_TOKEN, device=device)
 
         pca_state = load_pca_state_from_hf(token=HF_TOKEN, device=device)
         
@@ -201,9 +205,9 @@ def compile_model(should_compile):
     """Compile the model for faster inference."""
     global model, model_compiled, _model_compiled
     
-    # If on Zero GPU, compilation is not supported
-    if IS_ZEROGPU:
-        return gr.update(value=False, interactive=False), gr.update(value="⚠️ Compile disabled on Zero GPU", visible=True)
+    # Если компиляция недоступна (CPU или Zero GPU), отключаем чекбокс
+    if not COMPILE_AVAILABLE:
+        return gr.update(value=False, interactive=False), gr.update(value="⚠️ Compile disabled on CPU/Zero GPU", visible=True)
     
     if not should_compile:
         # User unchecked - clear status and allow toggling
@@ -221,9 +225,9 @@ def do_compile():
     """Actually perform the compilation by creating a separate compiled model."""
     global model, model_compiled, _model_compiled
     
-    # Skip if on Zero GPU
-    if IS_ZEROGPU:
-        return gr.update(value="⚠️ Compile disabled on Zero GPU", visible=True), gr.update(interactive=False)
+    # Если компиляция недоступна, пропускаем
+    if not COMPILE_AVAILABLE:
+        return gr.update(value="⚠️ Compile disabled on CPU/Zero GPU", visible=True), gr.update(interactive=False)
     
     if _model_compiled:
         return gr.update(value="", visible=False), gr.update(interactive=True)
@@ -324,6 +328,11 @@ def generate_audio(
     # Load models on first use (required for Zero GPU)
     load_models()
     
+    # Если компиляция недоступна, игнорируем флаг use_compile
+    if not COMPILE_AVAILABLE:
+        use_compile = False
+        # print("Compile not available on CPU/Zero GPU, using uncompiled model")
+    
     # Choose which model to use based on compile setting
     global model, model_compiled
     active_model = model_compiled if (use_compile and model_compiled is not None) else model
@@ -1855,13 +1864,13 @@ def init_and_compile():
     
     # Trigger compilation automatically on page load if not on Zero GPU
     # This ensures Simple mode (which defaults compile=True) gets compiled
-    if not IS_ZEROGPU:
+    if COMPILE_AVAILABLE:
         # Just call do_compile directly - it will load models and compile
         # Status updates will be visible in Advanced mode, hidden in Simple mode
         status_update, checkbox_update = do_compile()
         return session_id, status_update, checkbox_update
     else:
-        # On Zero GPU, don't try to compile
+        # On CPU or Zero GPU, don't try to compile
         return session_id, gr.update(), gr.update()
 
 SIMPLE_CSS = """
@@ -2299,13 +2308,13 @@ with gr.Blocks(title="Echo-TTS", css=LINK_CSS + SIMPLE_CSS, js=JS_CODE) as demo:
                     with gr.Column(scale=1, visible=False) as advanced_mode_compile_column:
                         compile_checkbox = gr.Checkbox(
                             label="Compile Model", 
-                            value=True,  # Default to True in simple mode
-                            interactive=not IS_ZEROGPU,
-                            info="Compile disabled on Zero GPU" if IS_ZEROGPU else "~20-30% faster after initial compilation"
+                            value=False,  # По умолчанию выключено для CPU
+                            interactive=COMPILE_AVAILABLE,
+                            info="Compile disabled on CPU/Zero GPU" if not COMPILE_AVAILABLE else "~20-30% faster after initial compilation"
                         )
                         compile_status = gr.Markdown(
-                            value="⚠️ Compile disabled on Zero GPU" if IS_ZEROGPU else "",
-                            visible=IS_ZEROGPU
+                            value="⚠️ Compile disabled on CPU/Zero GPU" if not COMPILE_AVAILABLE else "",
+                            visible=not COMPILE_AVAILABLE
                         )
                         use_custom_shapes_checkbox = gr.Checkbox(
                             label="Use Custom Shapes (Advanced)",