Skip to content

Commit

Permalink
Fix package requirements for Flux tool (bgruening#1550)
Browse files Browse the repository at this point in the history
* Change pytorch-cuda to pytorch-gpu to fix the not exist package
Update package requirements for Flux tool

* Update protobuf version requirement from 21.2 to 5.28.3

* Refactor flux.xml to use Docker container

* Update flux.xml to use python3 for script execution

* Update flux.xml

---------

Co-authored-by: Björn Grüning <[email protected]>
  • Loading branch information
arash77 and bgruening authored Nov 22, 2024
1 parent 4fecc1b commit f8d8a93
Showing 1 changed file with 4 additions and 13 deletions.
17 changes: 4 additions & 13 deletions tools/flux/flux.xml
Original file line number Diff line number Diff line change
Expand Up @@ -2,30 +2,21 @@
<description>text-to-image model</description>
<macros>
<token name="@TOOL_VERSION@">2024</token>
<token name="@VERSION_SUFFIX@">2</token>
<token name="@VERSION_SUFFIX@">3</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>
<requirement type="package" version="0.2.0">sentencepiece</requirement>
<requirement type="package" version="4.25.3">protobuf</requirement>
<requirement type="package" version="0.24.6">huggingface_hub</requirement>
<container type="docker">quay.io/galaxy/black_forest_labs_flux:2024-0</container>
</requirements>
<command detect_errors="exit_code"><![CDATA[
python '$__tool_directory__/flux.py'
python3 '$__tool_directory__/flux.py'
'$flux_models.fields.path'
'$input_type_selector'
'$prompt'
]]></command>
<configfiles>
</configfiles>
<inputs>
<param name="flux_models" label="Model data" type="select" help="contact the administrator of this Galaxy instance if you miss model data">
<param name="flux_models" label="Model data" type="select" help="Contact the administrator of our Galaxy instance if you miss model data">
<options from_data_table="huggingface">
<filter type="static_value" column="4" value="flux"/>
<filter type="static_value" column="5" value="1"/>
Expand Down

0 comments on commit f8d8a93

Please sign in to comment.