search

ToTheBeginning / PuLID

[NeurIPS 2024] Official code for PuLID: Pure and Lightning ID Customization via Contrastive Alignment
Apache License 2.0
2.69k stars 189 forks source link

After downloading flux a ( Command + C ) is being triggered #107

Closed kratos606 closed 1 month ago

kratos606 commented 1 month ago

Downloading shards: 100% 2/2 [00:52<00:00, 26.15s/it] Loading checkpoint shards: 100% 2/2 [00:00<00:00, 2.11it/s] tokenizer_config.json: 100% 905/905 [00:00<00:00, 7.20MB/s] vocab.json: 100% 961k/961k [00:00<00:00, 2.95MB/s] merges.txt: 100% 525k/525k [00:00<00:00, 2.24MB/s] special_tokens_map.json: 100% 389/389 [00:00<00:00, 3.20MB/s] tokenizer.json: 100% 2.22M/2.22M [00:00<00:00, 5.58MB/s] config.json: 100% 4.52k/4.52k [00:00<00:00, 23.5MB/s] model.safetensors: 100% 1.71G/1.71G [00:07<00:00, 225MB/s] Init model flux-dev-fp8.safetensors: 100% 11.9G/11.9G [03:20<00:00, 59.3MB/s] flux_dev_quantization_map.json: 100% 26.5k/26.5k [00:00<00:00, 49.8MB/s] ^C

kratos606 commented 1 month ago

The issue arises because Google Colab lacks sufficient resources to run PULID-FLUX effectively. You could try using Lightning AI for better support. If you're using 16GB memory, an L40s GPU should be sufficient; otherwise, you'll likely need at least an A100 GPU to make it work.