light tuning

Torta de luz

? CLI para ajustar los modelos Turbo GPT-3.5 rápidamente.

Concepto

LightTuning es una forma rápida de ajustar los modelos Turbo GPT-3.5. Escalar y mejorar un modelo de una pequeña conversación a un modelo ligero GPT-3.5 que puede tener un rendimiento superior a GPT-4. ¿Qué tan genial es eso?

? ‍ Uso

Instalación

Instale los paquetes requeridos con PIP.

pip3 install -r requirements.txt

Ejecute el script main.py para iniciar la herramienta.

python3 main.py

Será recibido con un mensaje de bienvenida y una lista de acciones para elegir:

• Generar conjunto de datos
• Subir el conjunto de datos
• Modelo de sintonía
• Todo

Seleccione una acción usando las teclas de flecha y presione Entrar.

Generar conjunto de datos

Si elige 'Generar DataSet', se le pedirá que proporcione el camino a la conversación de semillas. Si no proporciona una ruta, LightTuning le pedirá que ingrese.

Agregue una conversación de semilla al archivo. La conversación debe estar en el siguiente formato de conversación de OpenAI:

[
  {
    "role" : " system " ,
    "content" : " You are an assistant that can translate from English to Spanish. "
  },
  {
    "role" : " user " ,
    "content" : " Hello "
  },
  {
    "role" : " assistant " ,
    "content" : " Hola "
  },
  {
    "role" : " user " ,
    "content" : " Happy "
  },
  {
    "role" : " assistant " ,
    "content" : " Feliz "
  }
]

Subir el conjunto de datos

Si elige 'Cargar DataSet', se le pedirá que ingrese la ruta al archivo de conjunto de datos. Si no proporciona una ruta, la herramienta le pedirá que la ingrese. Luego se le pedirá que confirme la carga.

Modelo de sintonía

Si elige 'modelo de ajuste fino', se le pedirá que proporcione la ID del archivo para ajustar. Si no proporciona una ID de archivo, la herramienta le pedirá que lo ingrese.

Todo

Si elige 'All', la herramienta ejecutará todas las acciones anteriores en orden.

? Licencia

Distribuido bajo la licencia MIT. Vea LICENSE para más información.