| """ | |
| 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", | |
| ] | |