jina-ai / dalle-flow

🌊 A Human-in-the-Loop workflow for creating HD images from text
grpcs://dalle-flow.dev.jina.ai
2.83k stars 209 forks source link

Add another environmental variable so you can set CUDA_VISIBLE_DEVICES in flow_parser.py and docker image #105

Closed AmericanPresidentJimmyCarter closed 1 year ago

AmericanPresidentJimmyCarter commented 1 year ago

Right now it only uses a single GPU because CUDA_VISIBLE_DEVICES: 0 is set by default.