文件历史

30 次代码提交

作者 SHA1 备注 提交日期
Adrian (Mircea) Nenu e5d960c41a Update README.md for converting-models-to-coreml reference
Fix typo that does not anchor the links correctly
2025-01-23 08:28:43 -08:00
NehalBhandari f8e36141a9 Add support for mixed precision activation quantization for UNet model (#365) 2024-11-15 16:05:22 -08:00
Toby Roseman 7505f91208 Add option for sequential unet predictions 2024-10-23 14:23:19 -07:00
Toby Roseman e497107793 Allow not using classifier free guidance 2024-10-04 11:10:18 -07:00
LamTrinh.Dev cf4e1e8d6e Update README.md
Correct wording.
2024-08-11 20:07:36 +07:00
Zach Nagengast c891f4357b Add SD3 Pipeline (#329)
* Add SD3 Pipeline

Co-authored-by: atiorh <atiorh@users.noreply.github.com>
Co-authored-by: arda-argmax <arda-argmax@users.noreply.github.com>

* Use swift-transformers for tokenization

* Use diffusionkit converters in torch2coreml

* Documentation and cleanup

* Add model link

* Consolidate batch prediction logic

* Remove DecoderSD3.swift and consolidate logic into Decoder.swift

* Remove DiffusionKit MLX inference reference from README

---------

Co-authored-by: atiorh <atiorh@users.noreply.github.com>
Co-authored-by: arda-argmax <arda-argmax@users.noreply.github.com>
Co-authored-by: atila <atiorh@icloud.com>
2024-07-22 21:16:50 -07:00
Toby Roseman ec09fe064f code review feedback 2024-01-23 16:41:35 -08:00
Toby Roseman a9b84c658e Update Documentation 2024-01-23 08:32:22 -08:00
Toby Roseman 46c94e8a86 Link to Core ML documentation about why compiled models are faster 2023-11-07 12:46:13 -08:00
Toby Roseman 4c54204bb0 Compiled Models in Python 2023-11-07 12:13:31 -08:00
Atila Orhon 96c044e07b Add iPhone 12 Pro Benchmark 2023-09-28 14:34:09 -07:00
Atila Orhon 081a8cb2cc v1.1 2023-09-28 07:26:41 -07:00
Zach Nagengast 94dfc6b548 Support for SDXL refiner (#227)
* Initial support for SDXL refiner

* Cleanup

* Add arg for converting Unet in float32 precision

* Setup scale factor with pipeline in CLI

* Update cli arg and future warning

* Bundle refiner unet if specified

* Update script for bundled refiner

- Also skip loading model if check_output_correctness is missing, since the model does not require inferencing at conversion time

* Flip skip_model_load bool

* Cleanup

* Support bundled UnetRefiner

* Add seperate refiner config value

- Includes unloading base unet when swapping to refiner

* Update readme for SDXL refiner

* Add condition for new SDXL coreml input features

* Revert pipeline interface change, add extra logging on pipe load

* Reset model_version after refiner conversion

* Reset model_version before refiner conversion but after pipe init

* Add refiner chunking

* Ensure unets are unloaded for reduceMemory true

* Handle missing UnetRefiner.mlmodelc on pipeline load

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Prewarm refiner on load, unload on complete

* Force cpu_and_gpu for VAE until it can be fixed

* Include output dtype of np.float32 for all conversions

* Allow a custom VAE to be converted.

* Revert hardcoded reduceMemory

* Fix merge

* Default chunking arg for --merge-chunks-in-pipeline-model when called from torch2coreml

---------

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2023-09-26 13:09:58 -07:00
Atila Orhon a56e10232d Fixes #243
Co-authored-by: TBPer <thanawat.boonmee@gmail.com>
2023-08-30 14:02:20 -07:00
Mustafa Ateş Uzun 599bd577ed fix: read.md word spelling (#234) 2023-08-20 23:56:23 +03:00
Atila Orhon a57d127606 Mixed-bit Palettization
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2023-07-27 09:07:44 -10:00
Pedro Cuenca 48f07f2489 Links to quantized models in the Hugging Face hub. (#193) 2023-06-14 09:03:59 -07:00
Atila Orhon 7f9c58a6c1 WWDC23 update
Co-authored-by: davidfindlay1 <david.arthur.findlay@icloud.com>
2023-06-14 08:27:02 -07:00
Alejandro Isaza 940dba02ee Require Swift 5.8 (#179) 2023-05-01 10:18:09 -07:00
Atila Orhon 8fd725764f Fixes #178 2023-05-01 09:52:58 -07:00
Atila Orhon 5d2744e382 Add ControlNet example 2023-04-18 16:52:52 -07:00
Chimme 7f65e1c84b Support ControlNet (#153)
* add controlnet tentatively

* add controlnet in python code

* implement swift part

* support 8-bit quantization

* add controlnet unload when reduce memory

* remove irrelevant changes

* add more description about controlnet option in swift

* fix some for pr and update README

* pre-allocate zero shapedArray + make multi-controlnet faster
2023-04-18 16:17:59 -07:00
Atila Orhon 827d017723 Update README with torchRNG guidance 2023-02-14 21:55:50 -08:00
Pedro Cuenca e4ca3ff86b Add link to HF app to the Example Swift App section. (#121) 2023-02-03 10:23:05 -08:00
Timothy Kautz 086cc5e04e Image2image - python (#115)
* Add Encoder model to torch2coreml for image2image

and later for in-paining

* diagonal test with randn

* Revert "diagonal test with randn"

This reverts commit 270afe1bb970c9413d3c3c9d09c0bd4f2dcbfa19.

* readme updates for encoder

* pr comments
2023-01-30 22:22:52 -08:00
Pedro Cuenca 877ccb9666 Add brief instructions to download weights from the Hub (#10)
* Add brief instructions to download weights from the Hub.

This should be convenient and time-saving for many users.

* Capitalize ORIGINAL.

* Expand git lfs details.

* Make download code robust to symlinks.

coremltools seems to not like symlinks.

* Add mention to demo app.

* Add steps to git lfs process.

* Minor text change.

* Apply suggestions from review.
2022-12-22 11:00:35 -08:00
Yaroslav Spirin 66dde8da13 Update README.md (#66) 2022-12-15 16:23:41 -08:00
Christian Clauss d6a54fc2b0 Fix typos: Successfully facilitate getting pipeline overridden (#30) 2022-12-14 10:42:27 -08:00
Atila Orhon a4238a3718 README improvements and reduceMemory option in Swift
Co-authored-by: msiracusa <msiracusa+github@gmail.com>
2022-12-14 08:23:55 -08:00
Atila Orhon 583cc04ece Initial commit
Co-authored-by: aseemw <aseem.elec@gmail.com>
Co-authored-by: msiracusa <msiracusa+github@gmail.com>
2022-11-30 13:41:05 -06:00