File size: 538 Bytes
1e68047 a4dcd96 1e68047 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
"""
SafeLLaVA Model for HuggingFace Hub
This model is based on LLaVA v1.5: https://github.com/haotian-liu/LLaVA
Licensed under Apache License 2.0
SafeLLaVA adds image safety classification capabilities to LLaVA.
"""
# Re-export classes from safellava package for HuggingFace auto_map
from safellava.model.language_model.safe_llava_llama import (
SafetyConfig,
SafeLlavaLlamaForCausalLM,
SafetyCausalLMOutputWithPast,
)
__all__ = [
"SafetyConfig",
"SafeLlavaLlamaForCausalLM",
"SafetyCausalLMOutputWithPast",
]
|