Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ coverage.xml
*.pt2
examples/torchtrt_aoti_example/torchtrt_aoti_example
CLAUDE.md
package-lock.json
# clangd compilation database (generated by `bazel run //:refresh_compile_commands`)
/compile_commands.json
/external/
Expand Down
4 changes: 2 additions & 2 deletions docker/ngc_test/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,6 @@ parameterized>=0.2.0
pytest>=9.0.3
pytest-xdist>=3.6.1
pyyaml
transformers>=5.4.0
transformers>=5.15.1
timm
# torchvision nvidia-modelopt should use the one already installed inside the dlfw docker container
# torchvision nvidia-modelopt should use the one already installed inside the dlfw docker container
4 changes: 2 additions & 2 deletions examples/apps/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ This demo showcases the Flux image generation model accelerated using Torch-Tens
1. Install the required dependencies:

```bash
pip install gradio==5.29.0 nvidia-modelopt==0.27.1 diffusers==0.33.1 accelerate==1.3.0
pip install gradio==5.29.0 nvidia-modelopt==0.27.1 diffusers==0.38.0 accelerate==1.3.0
```

## Usage
Expand Down Expand Up @@ -58,4 +58,4 @@ python flux_demo.py --low_vram_mode
You can combine these options as needed. For example:
```bash
python flux_demo.py --dtype fp8 --dynamic_shapes --low_vram_mode
```
```
6 changes: 3 additions & 3 deletions examples/distributed_inference/requirement.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
accelerate
transformers
diffusers
tensorrt-llm
transformers>=5.15.1
diffusers>=0.38.0
tensorrt-llm
4 changes: 2 additions & 2 deletions examples/dynamo/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
cupy==13.1.0
triton==2.3.0
diffusers==0.30.3
transformers>=5.4.0
diffusers==0.38.0
transformers>=5.15.1
matplotlib
pandas
huggingface_hub
Expand Down
2 changes: 1 addition & 1 deletion examples/dynamo/torch_export_flux_dev.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@

.. code-block:: python

pip install sentencepiece=="0.2.0" transformers=="4.48.2" accelerate=="1.3.0" diffusers=="0.32.2" protobuf=="5.29.3"
pip install sentencepiece=="0.2.0" transformers>="5.15.1" accelerate=="1.3.0" diffusers=="0.38.0" protobuf=="5.29.3"

There are different components of the ``FLUX.1-dev`` pipeline such as ``transformer``, ``vae``, ``text_encoder``, ``tokenizer`` and ``scheduler``. In this example,
we demonstrate optimizing the ``transformer`` component of the model (which typically consumes >95% of the e2e diffusion latency)
Expand Down
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ test = [

test-ext = [
"timm>=1.0.3",
"transformers>=5.4.0",
"transformers>=5.15.1",
"torchvision>=0.30.0.dev,<0.31.0",
"flashinfer-python; sys_platform == 'linux' and python_version >= '3.11' and python_version < '3.13'",
]
Expand Down Expand Up @@ -178,7 +178,7 @@ conflicts = [
],
]
constraint-dependencies = [
"transformers>=5.4.0",
"transformers>=5.15.1",
]

[tool.uv.sources]
Expand Down
2 changes: 1 addition & 1 deletion tests/modules/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
timm==0.9.12
transformers>=5.4.0
transformers>=5.15.1
4 changes: 2 additions & 2 deletions toolchains/jp_workspaces/test_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,5 @@ parameterized>=0.2.0
pytest>=9.0.3
pytest-xdist>=3.6.1
pyyaml
transformers>=5.4.0
# TODO: currently timm torchvision nvidia-modelopt does not have distributions for jetson
transformers>=5.15.1
# TODO: currently timm torchvision nvidia-modelopt does not have distributions for jetson
2 changes: 1 addition & 1 deletion tools/perf/Flux/create_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ pip install tensorrt==10.14.1.48 --force-reinstall
pip3 install --pre torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130


pip install sentencepiece=="0.2.0" transformers=="4.48.2" accelerate=="1.3.0" diffusers=="0.32.2" protobuf=="5.29.3"
pip install sentencepiece=="0.2.0" transformers>="5.15.1" accelerate=="1.3.0" diffusers=="0.38.0" protobuf=="5.29.3"

pip install notebook
pip install gradio safetensors peft pyinstrument
Expand Down
4 changes: 2 additions & 2 deletions tools/perf/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ argparse
pyyaml
onnx
pandas
transformers>=5.4.0
diffusers==0.34.0
transformers>=5.15.1
diffusers==0.38.0
timm==0.9.8
monai==1.6.0
Loading