Fix FLUX test with correct env var (NVIDIA-NeMo#14149)

* Fix FLUX test with correct env var

Signed-off-by: Ao Tang <aot@nvidia.com>

* Remove optional

Signed-off-by: Ao Tang <aot@nvidia.com>

---------

Signed-off-by: Ao Tang <aot@nvidia.com>
Signed-off-by: Amir Hussein <amhussein@nvidia.com>

Author: suiyoubi

.github/workflows/cicd-main-nemo2.yml

@@ -270,7 +270,6 @@ jobs:
             script: L2_NeMo_2_Diffusion_Taskencoder_Test
           - runner: self-hosted-azure
             script: L2_NeMo_2_Flux_Import_Test
-            is-optional: true
           - runner: self-hosted-azure
             script: L2_NeMo_2_Flux_Inference_Test
           - runner: self-hosted-azure

tests/functional_tests/L2_NeMo_2_Flux_Import_Test.sh

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-TRANSFORMER_OFFLINE=1 HF_HOME=/home/TestData/diffusion/ coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo tests/collections/diffusion/models/flux/test_flux_hf_importer.py
+TRANSFORMERS_OFFLINE=1 HF_HOME=/home/TestData/diffusion/ coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo tests/collections/diffusion/models/flux/test_flux_hf_importer.py

tests/functional_tests/L2_NeMo_2_Flux_Inference_Test.sh

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-TRANSFORMER_OFFLINE=1 MASTER_PORT=29500 MASTER_ADDR=127.0.0.1 LOCAL_RANK=0 HF_HOME=/home/TestData/diffusion coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo \
+TRANSFORMERS_OFFLINE=1 MASTER_PORT=29500 MASTER_ADDR=127.0.0.1 LOCAL_RANK=0 HF_HOME=/home/TestData/diffusion coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo \
   /workspace/scripts/flux/flux_controlnet_infer.py \
   --t5_version google/t5-v1_1-xxl \
   --t5_load_config_only \
@@ -25,7 +25,7 @@ TRANSFORMER_OFFLINE=1 MASTER_PORT=29500 MASTER_ADDR=127.0.0.1 LOCAL_RANK=0 HF_HO
   --flux_num_joint_layers 1 \
   --flux_num_single_layers 1
 
-TRANSFORMER_OFFLINE=1 MASTER_PORT=29500 MASTER_ADDR=127.0.0.1 LOCAL_RANK=0 HF_HOME=/home/TestData/diffusion coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo \
+TRANSFORMERS_OFFLINE=1 MASTER_PORT=29500 MASTER_ADDR=127.0.0.1 LOCAL_RANK=0 HF_HOME=/home/TestData/diffusion coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo \
   /workspace/scripts/flux/flux_infer.py \
   --t5_version google/t5-v1_1-xxl \
   --t5_load_config_only \

tests/functional_tests/L2_NeMo_2_Flux_Training_Test.sh

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-TRANSFORMER_OFFLINE=1 HF_HOME=/home/TestData/diffusion/ coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo tests/collections/diffusion/models/flux/test_flux_controlnet_training.py --yes -v
+TRANSFORMERS_OFFLINE=1 HF_HOME=/home/TestData/diffusion/ coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo tests/collections/diffusion/models/flux/test_flux_controlnet_training.py --yes -v
 
 
-TRANSFORMER_OFFLINE=1 HF_HOME=/home/TestData/diffusion/ coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo tests/collections/diffusion/models/flux/test_flux_controlnet_training.py --yes -v --factory flux_training
+TRANSFORMERS_OFFLINE=1 HF_HOME=/home/TestData/diffusion/ coverage run -a --data-file=/workspace/.coverage --source=/workspace/nemo tests/collections/diffusion/models/flux/test_flux_controlnet_training.py --yes -v --factory flux_training