acb20338b1
put HF_ENDPOINT into shared for #15443
2024-04-06 08:53:21 +03:00
73f7812045
Merge pull request #15443 from Satariall/add-hf_endpoint-variable
...
Use HF_ENDPOINT variable for HuggingFace domain with default
2024-04-06 08:48:55 +03:00
989b89b12a
Use HF_ENDPOINT variable for HuggingFace domain with default
...
Modified the list_models function to dynamically construct the model URL
by using an environment variable for the HuggingFace domain. This allows
for greater flexibility in specifying the domain and ensures that the
modification is also compatible with the Hub client library. By
supporting different environments or requirements without hardcoding the
domain name, this change facilitates the use of custom HuggingFace
domains not only within our code but also when interacting with the Hub
client library.
2024-04-05 13:02:49 +03:00
8bebfde701
Merge pull request #15350 from baseco/memory-bug-fix
...
minor bug fix of sd model memory management
2024-03-30 07:37:10 +03:00
f4633cb9c0
fix: when find already_loaded model, remove loaded by array index
2024-03-26 18:29:51 +08:00
f62217b65d
minor bug fix of sd model memory management
2024-03-25 10:38:15 -07:00
e3fa46f26f
Fix various typos with crate-ci/typos
2024-03-04 08:42:07 +02:00
141a17e969
style changes for #14979
2024-03-02 06:55:03 +03:00
da67afe5f6
call apply_alpha_schedule_override in load_model_weights for #14979
2024-03-02 06:54:58 +03:00
28bc85a20a
Merge pull request #14979 from drhead/refiner_cumprod_fix
...
Protect alphas_cumprod during refiner switchover
2024-03-02 06:54:55 +03:00
ee470cc6a3
style changes for #14979
2024-03-02 06:54:11 +03:00
1a51b166a0
call apply_alpha_schedule_override in load_model_weights for #14979
2024-03-02 06:53:53 +03:00
e2cd92ea23
move refiner fix to sd_models.py
2024-02-26 23:43:27 -05:00
a183de04e3
Execute model_loaded_callback after moving to target device
2024-01-06 20:03:33 +08:00
267fd5d76b
Merge pull request #14145 from drhead/zero-terminal-snr
...
Implement zero terminal SNR noise schedule option
2024-01-01 14:45:12 +03:00
672dc4efa8
Fix forced reload
2023-12-06 15:16:10 +08:00
78acdcf677
fix variable
2023-12-02 14:09:18 -05:00
dc1adeecdd
Create alphas_cumprod_original on full precision path
2023-12-02 14:06:56 -05:00
50a21cb09f
Ensure the cached weight will not be affected
2023-12-02 22:06:47 +08:00
110485d5bb
Merge branch 'dev' into test-fp8
2023-12-02 17:00:09 +08:00
6080045b2a
Add support for SD 2.1 Turbo, by converting the state dict from SGM to LDM on load
2023-12-01 22:58:05 -05:00
b25c126ccd
Protect alphas_cumprod from downcasting
2023-11-29 17:38:53 -05:00
40ac134c55
Fix pre-fp8
2023-11-25 12:35:09 +08:00
370a77f8e7
Option for using fp16 weight when apply lora
2023-11-21 19:59:34 +08:00
598da5cd49
Use options instead of cmd_args
2023-11-19 15:50:06 +08:00
cd12256575
Merge branch 'dev' into test-fp8
2023-11-16 21:53:13 +08:00
6ad666e479
more changes for #13865 : fix formatting, rename the function, add comment and add a readme entry
2023-11-05 19:46:20 +03:00
80d639a440
linter
2023-11-05 19:32:21 +03:00
ff805d8d0e
Merge branch 'dev' into master
2023-11-05 19:30:57 +03:00
44c5097375
Use devices.torch_gc() instead of empty_cache()
2023-11-05 20:31:57 +05:30
ff1609f91e
Add SSD-1B as a supported model
2023-11-05 19:13:49 +05:30
d4d3134f6d
ManualCast for 10/16 series gpu
2023-10-28 15:24:26 +08:00
dda067f64d
ignore mps for fp8
2023-10-25 19:53:22 +08:00
bf5067f50c
Fix alphas cumprod
2023-10-25 12:54:28 +08:00
4830b25136
Fix alphas_cumprod dtype
2023-10-25 11:53:37 +08:00
1df6c8bfec
fp8 for TE
2023-10-25 11:36:43 +08:00
9c1eba2af3
Fix lint
2023-10-24 02:11:27 +08:00
eaa9f5162f
Add CPU fp8 support
...
Since norm layer need fp32, I only convert the linear operation layer(conv2d/linear)
And TE have some pytorch function not support bf16 amp in CPU. I add a condition to indicate if the autocast is for unet.
2023-10-24 01:49:05 +08:00
5f9ddfa46f
Add sdxl only arg
2023-10-19 23:57:22 +08:00
7c128bbdac
Add fp8 for sd unet
2023-10-19 13:56:17 +08:00
282903bb67
repair unload sd checkpoint button
2023-10-15 09:41:02 +03:00
0619df9835
use shallow copy for #13535
2023-10-14 08:01:04 +03:00
7cc96429f2
Merge pull request #13535 from chu8129/dev
...
fix: checkpoints_loaded:{checkpoint:state_dict}, model.load_state_dict issue in dict value empty
2023-10-14 08:00:04 +03:00
770ee23f18
reverst
2023-10-07 15:38:50 +08:00
76010a51ef
up
2023-10-07 15:36:01 +08:00
951842d785
Merge pull request #13139 from AUTOMATIC1111/ckpt-dir-path-separator
...
fix `--ckpt-dir` path separator and option use `short name` for checkpoint dropdown
2023-09-30 10:02:28 +03:00
87b50397a6
add missing import, simplify code, use patches module for #13276
2023-09-30 09:11:31 +03:00
e309583f29
Merge pull request #13276 from woweenie/patch-1
...
patch DDPM.register_betas so that users can put given_betas in model yaml
2023-09-30 09:01:12 +03:00
8e355fbd75
fix
2023-09-18 16:45:42 +08:00
d9d94141dc
patch DDPM.register_betas so that users can put given_betas in model yaml
2023-09-15 18:59:44 +02:00
AUTOMATIC1111