Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from configs import T5_SHUNT_REPOS
|
|
14 |
# βββ Device & Model Setup βββββββββββββββββββββββββββββββββββββ
|
15 |
# Don't initialize CUDA here for ZeroGPU compatibility
|
16 |
device = None # Will be set inside the GPU function
|
17 |
-
dtype = torch.
|
18 |
|
19 |
# Don't load models here - will load inside GPU function
|
20 |
t5_tok = None
|
|
|
14 |
# βββ Device & Model Setup βββββββββββββββββββββββββββββββββββββ
|
15 |
# Don't initialize CUDA here for ZeroGPU compatibility
|
16 |
device = None # Will be set inside the GPU function
|
17 |
+
dtype = torch.float32
|
18 |
|
19 |
# Don't load models here - will load inside GPU function
|
20 |
t5_tok = None
|