29 Commits

Author	SHA1	Message	Date
Kohya S	8919b31145	use original ControlNet instead of Diffusers	2024-09-29 23:07:34 +09:00
Kohya S	9ddb4d7a01	update readme and help message etc.	2024-05-12 17:55:08 +09:00
Zovjsra	64916a35b2	add disable_mmap to args	2024-04-16 16:40:08 +08:00
Yuta Hayashibe	5f6bf29e52	Replace print with logger if they are logs (#905) ... * Add get_my_logger() * Use logger instead of print * Fix log level * Removed line-breaks for readability * Use setup_logging() * Add rich to requirements.txt * Make simple * Use logger instead of print --------- Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>	2024-02-04 18:14:34 +09:00
Kohya S	716bad188b	Update dependencies ref #1024	2024-01-04 19:53:25 +09:00
Kohya S	5713d63dc5	add temporary workaround for playground-v2	2023-12-06 23:08:02 +09:00
Kohya S	90c47140b8	add support model without position_ids	2023-09-13 17:59:34 +09:00
Kohya S	0ee75fd75d	fix typos, add comments etc.	2023-09-03 12:24:15 +09:00
Isotr0py	2e0942d5c8	delet missed line	2023-08-27 20:45:40 +08:00
Isotr0py	6155f9c171	intergrate fp16/bf16 to model loading	2023-08-27 19:16:23 +08:00
Kohya S	92f41f1051	update sdxl ver in lora metadata from v0-9 to v1-0	2023-08-06 22:06:48 +09:00
Kohya S	c142dadb46	support sai model spec	2023-08-06 21:50:05 +09:00
Kohya S	b62185b821	change method name, add comments	2023-07-30 13:34:07 +09:00
Isotr0py	d9180c03f6	fix typos for _load_state_dict	2023-07-29 22:25:00 +08:00
Isotr0py	1199eacb72	fix typo	2023-07-28 13:49:37 +08:00
Isotr0py	fdb58b0b62	fix mismatch dtype	2023-07-28 13:47:54 +08:00
Isotr0py	315fbc11e5	refactor model loading to catch error	2023-07-28 13:10:38 +08:00
Isotr0py	272dd993e6	Merge branch 'sdxl' into sdxl	2023-07-28 10:19:37 +08:00
Isotr0py	96a52d9810	add dtype to u-net loading	2023-07-27 23:58:25 +08:00
Isotr0py	eec6aaddda	fix safetensors error: device invalid	2023-07-23 13:29:29 +08:00
Isotr0py	bb167f94ca	init unet with empty weights	2023-07-23 13:17:11 +08:00
Kohya S	50b53e183e	re-organize import	2023-07-23 13:33:02 +09:00
Kohya S	8fa5fb2816	support diffusers format for SDXL	2023-07-12 21:57:14 +09:00
ykume	1d25703ac3	add generation script	2023-07-09 13:33:26 +09:00
Kohya S	4a34e5804e	fix to work with .ckpt from comfyui	2023-07-05 21:55:43 +09:00
Kohya S	3d0375daa6	fix to work sdxl state dict without logit_scale	2023-07-05 21:45:30 +09:00
Kohya S	a751dc25d6	use CLIPTextModelWithProjection	2023-06-27 20:48:06 +09:00
Kohya S	747af145ed	add sdxl fine-tuning and LoRA	2023-06-26 08:07:24 +09:00
Kohya S	f7f762c676	add minimal inference code for sdxl	2023-06-24 11:52:26 +09:00