Description
Ce projet implique le fine-tuning d’un modèle FLUX.1 [dev] en utilisant mon propre visage comme jeu de données. En exploitant la puissance de calcul d’un GPU A100, le modèle a été entraîné à reconnaître et à générer des sorties personnalisées basées sur les caractéristiques faciales.
Étapes clés :
- Préparation du jeu de données : Collecte et prétraitement d’images de mon visage pour créer un jeu de données d’entraînement.
- Fine-tuning du modèle : Ajustement fin du modèle FLUX.1 [dev] en utilisant le framework LoRa et AI-toolkit.
Ce projet démontre la puissance des modèles d’IA personnalisés et le gain de performance offert par du matériel haut de gamme comme le GPU A100.