ReferencePython SDK ReferenceNemoguardrailsLibraryJailbreak DetectionModel Based

nemoguardrails.library.jailbreak_detection.model_based.models

View as Markdown

Module Contents

Classes

Data

SNOWFLAKE_MODEL_ID

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'