make tf32

app.py

@@ -178,7 +178,8 @@ def run_edit(audio_file, caption, num_steps, guidance_scale, guidance_rescale, s
     from omegaconf import OmegaConf
     from safetensors.torch import load_file
     import diffusers.schedulers as noise_schedulers
-    
+    torch.backends.cuda.matmul.allow_tf32 = False
+    torch.backends.cudnn.allow_tf32 = False
 
 
 
@@ -330,8 +331,10 @@ def build_demo():
     return demo
 
 if __name__ == "__main__":
+    print("[BOOT] entering main()", flush=True)
     demo = build_demo()
     port = int(os.environ.get("PORT", "7860"))
+    print(f"[BOOT] launching gradio on 0.0.0.0:{port}", flush=True)
     demo.queue().launch(
         server_name="0.0.0.0",
         server_port=port,

requirements.txt

@@ -28,4 +28,4 @@ wandb
 tensorboard
 swanlab
 
-spaces>=0.30.0
+spaces