Spaces:
Sleeping
Sleeping
V2
Browse files
app.py
CHANGED
|
@@ -20,9 +20,8 @@ class OptimizedShukaASR:
|
|
| 20 |
model='sarvamai/shuka_v1',
|
| 21 |
trust_remote_code=True,
|
| 22 |
device=-1, # Force CPU
|
| 23 |
-
torch_dtype=torch.float16, # Use half precision
|
| 24 |
model_kwargs={
|
| 25 |
-
"torch_dtype": torch.
|
| 26 |
"low_cpu_mem_usage": True,
|
| 27 |
"use_cache": True,
|
| 28 |
}
|
|
|
|
| 20 |
model='sarvamai/shuka_v1',
|
| 21 |
trust_remote_code=True,
|
| 22 |
device=-1, # Force CPU
|
|
|
|
| 23 |
model_kwargs={
|
| 24 |
+
"torch_dtype": torch.float32, # Use float32 for CPU
|
| 25 |
"low_cpu_mem_usage": True,
|
| 26 |
"use_cache": True,
|
| 27 |
}
|