Acytel commited on
Commit
1656c8d
·
1 Parent(s): fb485e1

feat: mocked OnnxConfig

Browse files
Files changed (2) hide show
  1. api.py +1 -1
  2. bhashini.py +3 -0
api.py CHANGED
@@ -20,7 +20,7 @@ from neuro_symbolic import run_neuro_symbolic_pipeline
20
  from whatsapp.webhook import router as whatsapp_router
21
  from config import settings
22
 
23
- # --- SECURE KEYS --
24
  SUPABASE_URL = settings.SUPABASE_URL
25
  SUPABASE_KEY = settings.SUPABASE_KEY
26
  GROQ_API_KEY = settings.GROQ_API_KEY
 
20
  from whatsapp.webhook import router as whatsapp_router
21
  from config import settings
22
 
23
+ # --- SECURE KEYS ---
24
  SUPABASE_URL = settings.SUPABASE_URL
25
  SUPABASE_KEY = settings.SUPABASE_KEY
26
  GROQ_API_KEY = settings.GROQ_API_KEY
bhashini.py CHANGED
@@ -14,7 +14,10 @@ if 'transformers.onnx' not in sys.modules:
14
  mock_onnx = types.ModuleType('transformers.onnx')
15
  class MockOnnxConfig:
16
  pass
 
 
17
  mock_onnx.OnnxConfig = MockOnnxConfig
 
18
  sys.modules['transformers.onnx'] = mock_onnx
19
  # --------------------------------
20
 
 
14
  mock_onnx = types.ModuleType('transformers.onnx')
15
  class MockOnnxConfig:
16
  pass
17
+ class MockOnnxSeq2SeqConfigWithPast(MockOnnxConfig):
18
+ pass
19
  mock_onnx.OnnxConfig = MockOnnxConfig
20
+ mock_onnx.OnnxSeq2SeqConfigWithPast = MockOnnxSeq2SeqConfigWithPast
21
  sys.modules['transformers.onnx'] = mock_onnx
22
  # --------------------------------
23