happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
Fooocus/ldm_patched/modules
History
Hasham Vakani ⚡ 15547a8720 fix: Add NVIDIA Blackwell (RTX 50xx, sm_120) support 
- Use bfloat16 dtype for UNet on Blackwell GPUs (compute major >= 12)
  which have native bf16 tensor core support
- Skip manual_cast for bfloat16 weights to avoid unnecessary casting
- Fix numpy TypeError with bfloat16 tensors in patch.py and
  ip_adapter.py by converting to float32 before .numpy() calls

Tested on RTX 5070 (sm_120, CUDA 12.8) with PyTorch nightly (cu128).
Generates images at ~3.2 it/s including Image Prompt (IP-Adapter) mode.

Fixes #3862, #4123, #4141
 		2026-03-04 05:43:44 +05:00
..
sd1_tokenizer 2.1.826 2023-12-12 11:38:05 -08:00
args_parser.py feat: support download of huggingface files from a mirror website (#2637) 2024-04-06 15:25:19 +02:00
checkpoint_pickle.py 2.1.826 2023-12-12 11:38:05 -08:00
clip_config_bigg.json 2.1.826 2023-12-12 11:38:05 -08:00
clip_model.py backend 2024-01-27 05:12:34 -08:00
clip_vision.py backend 2024-01-27 05:12:34 -08:00
clip_vision_config_g.json 2.1.826 2023-12-12 11:38:05 -08:00
clip_vision_config_h.json 2.1.826 2023-12-12 11:38:05 -08:00
clip_vision_config_vitl.json 2.1.826 2023-12-12 11:38:05 -08:00
conds.py replaced the custom lcm function with math.lcm (#1122) 2024-02-10 19:28:10 +01:00
controlnet.py backend 2024-01-27 05:12:34 -08:00
diffusers_convert.py 2.1.826 2023-12-12 11:38:05 -08:00
diffusers_load.py backend 2024-01-27 05:12:34 -08:00
gligen.py backend 2024-01-27 05:12:34 -08:00
latent_formats.py feat: add support for playground v2.5 (#3073) 2024-06-04 20:15:49 +02:00
lora.py 2.1.826 2023-12-12 11:38:05 -08:00
model_base.py backend 2024-01-27 05:12:34 -08:00
model_detection.py backend 2024-01-27 05:12:34 -08:00
model_management.py fix: Add NVIDIA Blackwell (RTX 50xx, sm_120) support 2026-03-04 05:43:44 +05:00
model_patcher.py backend 2024-01-27 05:12:34 -08:00
model_sampling.py fix: correctly set alphas_cumprod (#3106) 2024-06-06 13:42:26 +02:00
ops.py backend 2024-01-27 05:12:34 -08:00
options.py 2.1.826 2023-12-12 11:38:05 -08:00
sample.py backend 2024-01-27 05:12:34 -08:00
samplers.py feat: add restart sampler (#3219) 2024-07-01 14:24:21 +02:00
sd.py feat: add optional model VAE select (#2867) 2024-05-09 18:59:35 +02:00
sd1_clip.py Use weights_only for loading (#3427) 2024-08-03 12:33:01 +02:00
sd1_clip_config.json 2.1.826 2023-12-12 11:38:05 -08:00
sd2_clip.py 2.1.826 2023-12-12 11:38:05 -08:00
sd2_clip_config.json 2.1.826 2023-12-12 11:38:05 -08:00
sdxl_clip.py 2.1.826 2023-12-12 11:38:05 -08:00
supported_models.py backend 2024-01-27 05:12:34 -08:00
supported_models_base.py backend 2024-01-27 05:12:34 -08:00
utils.py 2.1.826 2023-12-12 11:38:05 -08:00