You can deploy this model and start earning money today!

elidepb/
mermaid-tinylora

Name: mermaid-tinylora
Availability: InStock

safetensors

Views

No views yet

Mermaid-LM - Generador de Diagramas Mermaid

Modelo fine-tuned basado en TinyLlama para generar código Mermaid.

Información del Modelo

Modelo Base: TinyLlama/TinyLlama-1.1B-Chat-v1.0
Tipo: LoRA Fine-tuned
Tamaño: ~52.1 MB
Fecha de Creación: 2026-02-19 05:58:30

Diagramas Soportados

✅ Flowchart: Diagramas de flujo y procesos
✅ Mindmap: Mapas mentales jerárquicos
✅ Timeline: Líneas de tiempo y cronologías

Instalación

pip install torch transformers peft accelerate

Uso Rápido

python

from inference import MermaidGenerator

# Inicializar el generador
generator = MermaidGenerator(
    model_path="./",  # Ruta a este directorio
    base_model_name="TinyLlama/TinyLlama-1.1B-Chat-v1.0"
)

# Generar diagrama
mermaid_code = generator.generate_simple(
    "Create a flowchart for user authentication process"
)

print(mermaid_code)

Archivos Incluidos

adapter_model.bin - Pesos del modelo LoRA
adapter_config.json - Configuración LoRA
tokenizer/ - Tokenizer completo
inference.py - Clase para inferencia
config.py - Configuraciones del modelo
orchestrator.py - Orquestador de generación y corrección
mermaid_validator.py - Validador de sintaxis (Ink + Py)
model_info.json - Información detallada del modelo

Parámetros de Generación

max_new_tokens: 400 (por defecto)
temperature: 0.6
top_p: 0.85
top_k: 50

Ejemplo Avanzado