CARLEXsX commited on
Commit
f4ec834
·
verified ·
1 Parent(s): 8341849

Update flux_kontext_helpers.py

Browse files
Files changed (1) hide show
  1. flux_kontext_helpers.py +1 -1
flux_kontext_helpers.py CHANGED
@@ -94,5 +94,5 @@ class Generator:
94
  print("Inicializando o Compositor de Cenas (FluxKontext)...")
95
  hf_token = os.getenv('HF_TOKEN')
96
  if hf_token: huggingface_hub.login(token=hf_token)
97
- flux_kontext_singleton = Generator(device_id='cuda:0')
98
  print("Compositor de Cenas pronto.")
 
94
  print("Inicializando o Compositor de Cenas (FluxKontext)...")
95
  hf_token = os.getenv('HF_TOKEN')
96
  if hf_token: huggingface_hub.login(token=hf_token)
97
+ flux_kontext_singleton = Generator(device_ids=['cuda:0', 'cuda:1', 'cuda:2', 'cuda:3'])
98
  print("Compositor de Cenas pronto.")