From 61ef69066b5fb4ff882b0de4e14f54b97b384f2a Mon Sep 17 00:00:00 2001
From: Arash Kadkhodaei <arash77.kad@gmail.com>
Date: Wed, 30 Oct 2024 13:50:01 +0100
Subject: [PATCH] Add pytorch-cuda requirement to flux tool (#1538)

---
 tools/flux/flux.xml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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>