ComfyUI + Flux AI training(ai-toolkit)

dusty-nv / jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

MIT License

2.37k stars 482 forks source link

ComfyUI + Flux AI training(ai-toolkit) #594

Closed johnnynunez closed 3 months ago

johnnynunez commented 3 months ago

Credits: https://github.com/ostris/ai-toolkit, https://github.com/comfyanonymous/ComfyUI

johnnynunez commented 3 months ago

@dusty-nv I've updated Pytorch and their libraries, and opencv and bitsandbytes required for FLUX. Now pytorch has heterogeneus memory with +12.2 so you can use swap