Initial commit
This commit is contained in:
66
install_pix2struct.sh
Executable file
66
install_pix2struct.sh
Executable file
@@ -0,0 +1,66 @@
|
||||
#!/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 ""
|
||||
Reference in New Issue
Block a user