Add transformers restoration after ptq for vila

Signed-off-by: Yue <[email protected]>

Author: yueshen2016

examples/vlm_ptq/scripts/huggingface_example.sh

@@ -74,6 +74,10 @@ if [ -n "$KV_CACHE_QUANT" ]; then
 fi
 
 if [[ "${MODEL_NAME,,}" == *"vila"* ]]; then
+    # Save current transformers version for later restoration
+    ORIGINAL_TRANSFORMERS_VERSION=$(pip show transformers | grep Version | cut -d' ' -f2)
+    echo "Current transformers version: $ORIGINAL_TRANSFORMERS_VERSION"
+
     # Install required dependency for VILA
     pip install -r ../vlm_ptq/requirements-vila.txt
     # Clone original VILA repo
@@ -103,6 +107,13 @@ if [[ $TASKS =~ "quant" ]] || [[ ! -d "$SAVE_PATH" ]] || [[ ! $(ls -A $SAVE_PATH
     fi
 fi
 
+# Restore original transformers version immediately after PTQ for VILA models
+if [[ "${MODEL_NAME,,}" == *"vila"* ]] && [ -n "$ORIGINAL_TRANSFORMERS_VERSION" ]; then
+    echo "Restoring original transformers version: $ORIGINAL_TRANSFORMERS_VERSION"
+    pip install transformers==$ORIGINAL_TRANSFORMERS_VERSION
+    echo "Transformers version restored successfully."
+fi
+
 if [[ "$QFORMAT" != "fp8" ]]; then
     echo "For quant format $QFORMAT, please refer to the TensorRT-LLM documentation for deployment. Checkpoint saved to $SAVE_PATH."
     exit 0