27 Commits

Author	SHA1	Message	Date
Sean Sube	6d2dd0a043	apply lint	2023-03-05 07:20:43 -06:00
Sean Sube	10fbafaff0	fix(api): correct imports	2023-03-04 22:25:49 -06:00
Sean Sube	39d36618e6	feat(api): add support for custom tokens for textual inversions (#179)	2023-03-02 23:32:20 -06:00
Sean Sube	22e9d54408	apply lint	2023-03-02 22:32:37 -06:00
Sean Sube	be1c8cd79f	feat(api): support ckpt inversions by splitting multivector embeddings into multiple tokens	2023-03-02 22:11:27 -06:00
Sean Sube	9a0d2051fb	fix(api): better logging when converting textual inversions	2023-03-02 07:57:59 -06:00
Sean Sube	46aac263d5	feat(api): initial support for textual inversion embeddings from civitai and others (#179)	2023-03-01 19:09:51 -06:00
Sean Sube	4b77a00ca7	fix(api): load tokenizer with textual inversions	2023-03-01 19:08:31 -06:00
Sean Sube	21fc7c5968	fix(api): mark all convert methods as no_grad	2023-03-01 08:26:40 -06:00
Sean Sube	3f9f94fcb5	apply lint, remove unused	2023-02-28 23:05:17 -06:00
Sean Sube	2f4ab20f61	use filename for tensors	2023-02-28 22:49:53 -06:00
Sean Sube	74aae1b027	fix(api): write external weights into same directory as optimized model	2023-02-28 22:47:02 -06:00
Sean Sube	dbf9eaf1a4	fix(api): run shape inference before converting models to fp16 ... per discussion in https://github.com/microsoft/onnxruntime/issues/14827	2023-02-28 22:36:45 -06:00
Sean Sube	9ef89db8b0	extract tensors after conversion	2023-02-28 22:36:33 -06:00
Sean Sube	7e65e21410	reload model from proto file before converting	2023-02-28 22:36:26 -06:00
Sean Sube	2210ee849b	only convert inner nodes with ORT conversion helpers	2023-02-28 22:26:04 -06:00
Sean Sube	12fb7f52bb	fix(api): sanitize filenames in user input	2023-02-28 21:56:12 -06:00
Sean Sube	973b0a9352	add context arg	2023-02-25 12:13:22 -06:00
Sean Sube	d88f13cbd7	lint LoRA code and extras, replace public paths with context ones	2023-02-25 12:12:10 -06:00
Sean Sube	ce74183e97	blend LoRAs into existing model	2023-02-25 11:53:43 -06:00
Sean Sube	e8b5ff250d	add none option to inversion menu	2023-02-25 11:53:37 -06:00
Sean Sube	45f5fca383	use correct fn to make dirs	2023-02-25 11:53:29 -06:00
Sean Sube	3dfaef041e	fix devices, make subdir	2023-02-25 11:53:26 -06:00
Sean Sube	3f4b3fa322	load CLIP on training device	2023-02-25 11:53:23 -06:00
Sean Sube	6b4ced2608	load inversions from extras file	2023-02-25 11:53:20 -06:00
Sean Sube	9dedfc7b28	add experimental LoRA blender	2023-02-25 11:53:17 -06:00
Sean Sube	a31f7b9e1f	feat(api): convert Textual Inversion weights	2023-02-25 11:53:13 -06:00