nemoguardrails.library.jailbreak_detection.model_based.models
nemoguardrails.library.jailbreak_detection.model_based.models
This page is for version v0.24.1.
For other versions, use one of these documentation indexes:
- Latest (default): https://docs.nvidia.com/nemo/guardrails/latest/llms.txt
- v0.24.1: https://docs.nvidia.com/nemo/guardrails/v0.24.1/llms.txt
- v0.24.0: https://docs.nvidia.com/nemo/guardrails/v0.24.0/llms.txt
- v0.23.0: https://docs.nvidia.com/nemo/guardrails/v0.23.0/llms.txt
- v0.22.0: https://docs.nvidia.com/nemo/guardrails/v0.22.0/llms.txt
Module Contents
Classes
| Name | Description |
|---|---|
JailbreakClassifier | - |
SnowflakeEmbed | - |
Data
API
class nemoguardrails.library.jailbreak_detection.model_based.models.JailbreakClassifier(random_forest_path: str)
classifier
embed
= SnowflakeEmbed()
nemoguardrails.library.jailbreak_detection.model_based.models.JailbreakClassifier.__call__(text: str) -> typing.Tuple[bool, float]
class nemoguardrails.library.jailbreak_detection.model_based.models.SnowflakeEmbed()
device
= 'cuda' if torch.cuda.is_available() else 'cpu'
model
tokenizer
nemoguardrails.library.jailbreak_detection.model_based.models.SnowflakeEmbed.__call__(text: str)
nemoguardrails.library.jailbreak_detection.model_based.models.SNOWFLAKE_MODEL_ID = 'Snowflake/snowflake-arctic-embed-m-long'