diff --git a/tools/flux/flux.xml b/tools/flux/flux.xml
index c100a9cc7d..da0a6c909a 100644
--- a/tools/flux/flux.xml
+++ b/tools/flux/flux.xml
@@ -2,12 +2,13 @@
     <description>text-to-image model</description>
     <macros>
         <token name="@TOOL_VERSION@">2024</token>
-        <token name="@VERSION_SUFFIX@">1</token>
+        <token name="@VERSION_SUFFIX@">2</token>
     </macros>
     <requirements>
         <requirement type="package" version="3.12">python</requirement>
         <requirement type="package" version="2.4.1">pytorch</requirement>
         <requirement type="package" version="0.19.1">torchvision</requirement>
+        <requirement type="package" version="12.4">pytorch-cuda</requirement>
         <requirement type="package" version="0.30.2">diffusers</requirement>
         <requirement type="package" version="4.44.2">transformers</requirement>
         <requirement type="package" version="0.34.0">accelerate</requirement>