AvtnshM commited on
Commit
d0747b6
·
verified ·
1 Parent(s): 40d87de
Files changed (1) hide show
  1. app.py +1 -2
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.float16,
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
  }