Ryan Dick
|
35d8e6a05b
|
Tidy is_schnell detection logic.
|
2024-08-22 16:03:54 +00:00 |
|
Ryan Dick
|
afd4913a1b
|
Make FLUX get_noise(...) consistent across devices/dtypes.
|
2024-08-22 15:56:30 +00:00 |
|
Ryan Dick
|
6afb1be872
|
Mark FLUX nodes as prototypes.
|
2024-08-22 15:29:59 +00:00 |
|
Brandon Rising
|
3d251b4b93
|
Run ruff
|
2024-08-21 15:37:27 -04:00 |
|
Brandon Rising
|
fc760d3719
|
Remove automatic install of models during flux model loader, remove no longer used import function on context
|
2024-08-21 15:34:47 -04:00 |
|
Ryan Dick
|
0ce9faae7d
|
Fix max_seq_len field description.
|
2024-08-21 19:17:39 +00:00 |
|
maryhipp
|
25482c031b
|
fix(worker) fix T5 type
|
2024-08-21 14:45:46 -04:00 |
|
maryhipp
|
2c72295b1c
|
update flux_model_loader node to take a T5 encoder from node field instead of hardcoded list, assume all models have been downloaded
|
2024-08-21 14:27:34 -04:00 |
|
Brandon Rising
|
ada483f65e
|
Various styling and exception type updates
|
2024-08-21 11:59:04 -04:00 |
|
Ryan Dick
|
e680cf76f6
|
Address minor review comments.
|
2024-08-21 13:45:22 +00:00 |
|
Ryan Dick
|
253b2b1dc6
|
Rename t5Encoder -> t5_encoder.
|
2024-08-21 13:27:54 +00:00 |
|
Brandon Rising
|
c819da8859
|
Some cleanup of the tags and description of flux nodes
|
2024-08-21 09:11:15 -04:00 |
|
Brandon Rising
|
dd24f83d43
|
Fix styling/lint
|
2024-08-21 09:10:22 -04:00 |
|
Brandon Rising
|
da766f5a7e
|
Fix support for 8b quantized t5 encoders, update exception messages in flux loaders
|
2024-08-21 09:10:22 -04:00 |
|
Ryan Dick
|
5e2351f3bf
|
Fix FLUX output image clamping. And a few other minor fixes to make inference work with the full bfloat16 FLUX transformer model.
|
2024-08-21 09:10:22 -04:00 |
|
Brandon Rising
|
d705c3cf0e
|
Select dev/schnell based on state dict, use correct max seq len based on dev/schnell, and shift in inference, separate vae flux params into separate config
|
2024-08-21 09:10:20 -04:00 |
|
Brandon Rising
|
115f350f6f
|
Install sub directories with folders correctly, ensure consistent dtype of tensors in flux pipeline and vae
|
2024-08-21 09:09:39 -04:00 |
|
Brandon Rising
|
be6cb2c07c
|
Working inference node with quantized bnb nf4 checkpoint
|
2024-08-21 09:09:39 -04:00 |
|
Brandon Rising
|
4fb5529493
|
Remove unused param on _run_vae_decoding in flux text to image
|
2024-08-21 09:09:39 -04:00 |
|
Brandon Rising
|
b43ee0b837
|
Add nf4 bnb quantized format
|
2024-08-21 09:09:39 -04:00 |
|
Brandon Rising
|
3312fe8fc4
|
Run ruff, setup initial text to image node
|
2024-08-21 09:09:39 -04:00 |
|
Brandon Rising
|
01a2449dae
|
Add backend functions and classes for Flux implementation, Update the way flux encoders/tokenizers are loaded for prompt encoding, Update way flux vae is loaded
|
2024-08-21 09:09:37 -04:00 |
|
Brandon Rising
|
cfe9d0ce0a
|
Some UI cleanup, regenerate schema
|
2024-08-21 09:08:22 -04:00 |
|
Brandon Rising
|
46b6314482
|
Run Ruff
|
2024-08-21 09:06:38 -04:00 |
|
Brandon Rising
|
46d5107ff1
|
Run Ruff
|
2024-08-21 09:06:38 -04:00 |
|
Brandon Rising
|
6ea1278d22
|
Manage quantization of models within the loader
|
2024-08-21 09:06:34 -04:00 |
|
Brandon Rising
|
f425d3aa3c
|
Setup flux model loading in the UI
|
2024-08-21 09:04:37 -04:00 |
|
Ryan Dick
|
d7a39a4d67
|
WIP on moving from diffusers to FLUX
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
0e96794c6e
|
LLM.int8() quantization is working, but still some rough edges to solve.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
23a7328a66
|
Clean up NF4 implementation.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
c3cf8c3b6b
|
NF4 inference working
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
3ba60e1656
|
Split a FluxTextEncoderInvocation out from the FluxTextToImageInvocation. This has the advantage that we benfit from automatic caching when the prompt isn't changed.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
cdd47b657b
|
Make quantized loading fast for both T5XXL and FLUX transformer.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
e8fb8f4d12
|
Make float16 inference work with FLUX on 24GB GPU.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
9381211508
|
Add support for 8-bit quantizatino of the FLUX T5XXL text encoder.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
8cce4a40d4
|
Make 8-bit quantization save/reload work for the FLUX transformer. Reload is still very slow with the current optimum.quanto implementation.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
4833746698
|
Minor improvements to FLUX workflow.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
8b9bf55bba
|
Got FLUX schnell working with 8-bit quantization. Still lots of rough edges to clean up.
|
2024-08-21 08:59:19 -04:00 |
|
Ryan Dick
|
7b199fed4f
|
Use the FluxPipeline.encode_prompt() api rather than trying to run the two text encoders separately.
|
2024-08-21 08:59:18 -04:00 |
|
Ryan Dick
|
13513465c8
|
First draft of FluxTextToImageInvocation.
|
2024-08-21 08:59:18 -04:00 |
|
blessedcoolant
|
4f8a4b0f22
|
Merge branch 'main' into depth_anything_v2
|
2024-08-03 00:38:57 +05:30 |
|
Ryan Dick
|
981475a624
|
Merge branch 'main' into ryan/grounded-sam
|
2024-08-01 20:30:35 +02:00 |
|
Ryan Dick
|
27ac61a4fb
|
Expose all model options in the GroundingDinoInvocation and the SegmentAnythingInvocation.
|
2024-08-01 14:23:32 -04:00 |
|
Ryan Dick
|
675ffc2757
|
Remove BoundingBoxInvocation field name overrides.
|
2024-08-01 14:05:44 -04:00 |
|
Ryan Dick
|
44b21f10f1
|
Add a pydantic model_validator to BoundingBoxField to check the validity of the coords.
|
2024-08-01 14:00:57 -04:00 |
|
Ryan Dick
|
c6d49e8b1f
|
Shorten SegmentAnythingInvocation and GroundingDinoInvocatino docstrings, since they are used as the invocation descriptions in the UI.
|
2024-08-01 10:17:42 -04:00 |
|
Ryan Dick
|
e6a512aa86
|
(minor) Tweak order of mask operations.
|
2024-08-01 10:12:24 -04:00 |
|
Ryan Dick
|
c3a6a6fb22
|
Rename SegmentAnythingModelInvocation -> SegmentAnythingInvocation.
|
2024-08-01 10:00:36 -04:00 |
|
Ryan Dick
|
b9dc3460ba
|
Rename SegmentAnythingModel -> SegmentAnythingPipeline.
|
2024-08-01 09:57:47 -04:00 |
|
blessedcoolant
|
f5cfdcf32d
|
feat: Add BoundingBox Primitive Node
|
2024-08-01 04:09:08 +05:30 |
|