Staticaliza commited on
Commit
8c43d1b
·
verified ·
1 Parent(s): ad81304

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -1,5 +1,17 @@
1
  import gradio as gr
2
- import torch, numpy as np, spaces
 
 
 
 
 
 
 
 
 
 
 
 
3
  from zonos.model import Zonos
4
  from zonos.conditioning import make_cond_dict
5
 
 
1
  import gradio as gr
2
+ import os
3
+
4
+ os.environ["TORCH_COMPILE_DISABLE"] = "1" # disable torch.compile globally
5
+ os.environ["TORCHINDUCTOR_DISABLE"] = "1" # belt-and-suspenders for inductor
6
+
7
+ import torch, torch._dynamo # must come *after* the env vars
8
+ torch._dynamo.config.suppress_errors = True # swallow any stray compile calls
9
+ torch._dynamo.disable()
10
+
11
+ import spaces
12
+ import torch
13
+ import numpy as np
14
+
15
  from zonos.model import Zonos
16
  from zonos.conditioning import make_cond_dict
17