67 lines
1.6 KiB
Bash
Executable File
67 lines
1.6 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Script d'installation de Pix2Struct
|
|
# Installe les dépendances nécessaires pour le modèle Pix2Struct
|
|
#
|
|
|
|
set -e
|
|
|
|
echo "============================================================"
|
|
echo "INSTALLATION PIX2STRUCT"
|
|
echo "============================================================"
|
|
echo ""
|
|
|
|
# Couleurs
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
NC='\033[0m'
|
|
|
|
print_status() {
|
|
echo -e "${GREEN}[✓]${NC} $1"
|
|
}
|
|
|
|
print_info() {
|
|
echo -e "${YELLOW}[i]${NC} $1"
|
|
}
|
|
|
|
# Vérifier l'environnement virtuel
|
|
if [ ! -d "geniusia2/venv" ]; then
|
|
echo "Erreur: Environnement virtuel non trouvé"
|
|
echo "Exécute d'abord: python3 -m venv geniusia2/venv"
|
|
exit 1
|
|
fi
|
|
|
|
print_status "Environnement virtuel trouvé"
|
|
|
|
# Activer le venv
|
|
source geniusia2/venv/bin/activate
|
|
|
|
# Installer les dépendances
|
|
print_info "Installation de transformers>=4.35.0..."
|
|
pip install 'transformers>=4.35.0' --quiet
|
|
|
|
print_info "Installation de sentencepiece..."
|
|
pip install sentencepiece --quiet
|
|
|
|
print_status "Dépendances Pix2Struct installées"
|
|
|
|
# Vérifier l'installation
|
|
echo ""
|
|
print_info "Vérification de l'installation..."
|
|
|
|
if python -c "from transformers import Pix2StructProcessor, Pix2StructForConditionalGeneration" 2>/dev/null; then
|
|
print_status "Pix2Struct est prêt à l'emploi"
|
|
else
|
|
echo "Erreur: Impossible d'importer Pix2Struct"
|
|
exit 1
|
|
fi
|
|
|
|
echo ""
|
|
echo "============================================================"
|
|
print_status "Installation terminée avec succès!"
|
|
echo "============================================================"
|
|
echo ""
|
|
print_info "Pour utiliser Pix2Struct, configure:"
|
|
echo " embedding_model='pix2struct'"
|
|
echo ""
|