Put Flux inside jetson container

[martincerven]

Put content from https://www.jetson-ai-lab.com/tutorial_comfyui_flux.html

into jetson containers

[johnnynunez]

I will work on that

[johnnynunez]

training: #594

[dusty-nv]

Johhny is on it 😀 thanks Johhny!

[johnnynunez]

Put content from https://www.jetson-ai-lab.com/tutorial_comfyui_flux.html

into jetson containers

If you want testing: #594

for me is working now:

[johnnynunez]

CUDA_VERSION=12.2 jetson-containers build comfyui
sudo jetson-containers run -it comfyui:r36.3.0-cu122

[devrosx]

hello great :) its working on jetson xavier AGX ? tried to build manually but i have still this error module ‘torch’ has no attribute ‘float8_e4m3fn’

[johnnynunez]

hello great :) its working on jetson xavier AGX ? tried to build manually but i have still this error module ‘torch’ has no attribute ‘float8_e4m3fn’

I don't know. ComfyUI is very recently and works with latest version of pytorch.
But this error is more on the pytorch side:
comfyanonymous/ComfyUI#4297

[devrosx]

on AGX Xavier is latest CUDA 11.8 and torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
I would probably need a newer version to get rid of this error.