Write the generated images using the image crate. (#363)

* Use the image crate to write the generated images.

* Make the dependency optional.
This commit is contained in:
Laurent Mazare
2023-08-09 16:26:44 +02:00
committed by GitHub
parent be21d7e75a
commit 3a62aee91f
4 changed files with 31 additions and 6 deletions

View File

@ -33,6 +33,7 @@ cudarc = { version = "0.9.13", features = ["f16"] }
gemm = { version = "0.15.5", package = "candle-gemm" }
hf-hub = "0.2.0"
half = { version = "2.3.1", features = ["num-traits", "rand_distr"] }
image = { version = "0.24.7", default-features = false, features = ["jpeg", "png"] }
intel-mkl-src = { version = "0.8.1", features = ["mkl-static-lp64-iomp"] }
libc = { version = "0.2.147" }
log = "0.4"