Solução otimizada para operação de dispositivo de memória gráfica baixa
Para o problema de memória de vídeo insuficiente, é recomendável usarabordagem em fases::
- Certifique-se de ativar a desinstalação da CPU ao instalar: adicionar ao código
pipe.enable_model_cpu_offload()
- Usando o modo de meia precisão: Configuração
torch_dtype=torch.bfloat16
Reduz o espaço ocupado pela memória de vídeo do 40% - Reduza a resolução de saída: use temporariamente 512×512 para testar o efeito
- Renderização em pedaços com interface de usuário confortável: segmentação de imagens com fluxos de trabalho nodais
Mesmo com uma placa de vídeo RTX 2060 (6 GB), a geração de imagens ainda pode ser realizada pelo método acima, com um tempo de geração único de cerca de 3 a 5 minutos.
Essa resposta foi extraída do artigoFLUX.1 Krea: uma ferramenta gratuita de código aberto para gerar imagens altamente realistasO