Add pytorch-cuda requirement to flux tool (bgruening#1538)

planemo-autoupdate · Oct 30, 2024 · 61ef690 · 61ef690
1 parent 0959e15
commit 61ef690
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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>