Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
rohankaran
commited on
Commit
•
4962c42
1
Parent(s):
13d0c78
Activate additional models in guardrails_models.py
Browse filesThe code updates in guardrails_models.py have uncommented previously disabled models. By enabling these models in the 'get_all_models' function, we are enhancing our predictive system's versatility and allowing it to utilize additional language models such as "gpt3.5-turbo-1106", "Llama-2-70b-chat-hf", "Mixtral-8x7B-Instruct-v0.1", and various combinations with guardrails.
- guardrails_models.py +40 -40
guardrails_models.py
CHANGED
@@ -393,50 +393,50 @@ def gemini_pro_nemoguardrails(
|
|
393 |
|
394 |
def get_all_models():
|
395 |
return [
|
396 |
-
|
397 |
-
|
398 |
-
|
399 |
-
|
400 |
-
|
401 |
-
|
402 |
-
|
403 |
-
|
404 |
-
|
405 |
-
|
406 |
-
|
407 |
-
|
408 |
{
|
409 |
"name": "Gemini-Pro",
|
410 |
"model": gemini_pro,
|
411 |
},
|
412 |
-
|
413 |
-
|
414 |
-
|
415 |
-
|
416 |
-
|
417 |
-
|
418 |
-
|
419 |
-
|
420 |
-
|
421 |
-
|
422 |
-
|
423 |
-
|
424 |
-
|
425 |
-
|
426 |
-
|
427 |
-
|
428 |
-
|
429 |
-
|
430 |
-
|
431 |
-
|
432 |
-
|
433 |
-
|
434 |
-
|
435 |
-
|
436 |
-
|
437 |
-
|
438 |
-
|
439 |
-
|
440 |
{
|
441 |
"name": "Gemini-Pro + NeMo Guardrails",
|
442 |
"model": gemini_pro_nemoguardrails,
|
|
|
393 |
|
394 |
def get_all_models():
|
395 |
return [
|
396 |
+
{
|
397 |
+
"name": "gpt3.5-turbo-1106",
|
398 |
+
"model": gpt35_turbo,
|
399 |
+
},
|
400 |
+
{
|
401 |
+
"name": "Llama-2-70b-chat-hf",
|
402 |
+
"model": llama70B,
|
403 |
+
},
|
404 |
+
{
|
405 |
+
"name": "Mixtral-8x7B-Instruct-v0.1",
|
406 |
+
"model": mixtral7x8,
|
407 |
+
},
|
408 |
{
|
409 |
"name": "Gemini-Pro",
|
410 |
"model": gemini_pro,
|
411 |
},
|
412 |
+
{
|
413 |
+
"name": "gpt3.5-turbo-1106 + Llama Guard",
|
414 |
+
"model": gpt35_turbo_llamaguard,
|
415 |
+
},
|
416 |
+
{
|
417 |
+
"name": "Llama-2-70b-chat-hf + Llama Guard",
|
418 |
+
"model": llama70B_llamaguard,
|
419 |
+
},
|
420 |
+
{
|
421 |
+
"name": "Mixtral-8x7B-Instruct-v0.1 + Llama Guard",
|
422 |
+
"model": mixtral7x8_llamaguard,
|
423 |
+
},
|
424 |
+
{
|
425 |
+
"name": "Gemini-Pro + Llama Guard",
|
426 |
+
"model": gemini_pro_llamaguard,
|
427 |
+
},
|
428 |
+
{
|
429 |
+
"name": "gpt3.5-turbo-1106 + NeMo Guardrails",
|
430 |
+
"model": gpt35_turbo_nemoguardrails,
|
431 |
+
},
|
432 |
+
{
|
433 |
+
"name": "Llama-2-70b-chat-hf + NeMo Guardrails",
|
434 |
+
"model": llama70B_nemoguardrails,
|
435 |
+
},
|
436 |
+
{
|
437 |
+
"name": "Mixtral-8x7B-Instruct-v0.1 + NeMo Guardrails",
|
438 |
+
"model": mixtral7x8_nemoguardrails,
|
439 |
+
},
|
440 |
{
|
441 |
"name": "Gemini-Pro + NeMo Guardrails",
|
442 |
"model": gemini_pro_nemoguardrails,
|