[QEff.Finetuning] Adding different configs for functional and assert tests.

scripts/Jenkinsfile

@@ -25,7 +25,6 @@ pipeline {
                    pip install junitparser pytest-xdist &&
                    pip install librosa==0.10.2 soundfile==0.13.1 && #packages needed to load example for whisper testing
                    pip install --extra-index-url https://download.pytorch.org/whl/cpu timm==1.0.14 torchvision==0.22.0+cpu einops==0.8.1 && #packages to load VLMs
-                   pip install /opt/qti-aic/integrations/torch_qaic/py310/torch_qaic-0.1.0-cp310-cp310-linux_x86_64.whl && # For finetuning tests
                    rm -rf QEfficient"
                '''
            }
@@ -166,8 +165,9 @@ pipeline {
                 timeout(time: 5, unit: 'MINUTES') {
                     sh '''
                     sudo docker exec ${BUILD_TAG} bash -c "
+                    source /opt/torch-qaic-env/bin/activate &&
                     cd /efficient-transformers &&
-                    . preflight_qeff/bin/activate &&
+                    pip install .[test] &&
                     mkdir -p $PWD/cli_qaic_finetuning &&
                     export TOKENIZERS_PARALLELISM=false &&
                     export QEFF_HOME=$PWD/cli_qaic_finetuning &&

tests/finetune/constants.py

@@ -6,5 +6,5 @@
 # -----------------------------------------------------------------------------
 
 # Finetuning Test Constants
-LOSS_ATOL = 1e-3
-METRIC_ATOL = 1e-3
+LOSS_ATOL = 2e-2
+METRIC_ATOL = 3e-2