diffusers

Commit Graph

Author	SHA1	Message	Date
Kashif Rasul	b34be039f9	Karras VE, DDIM and DDPM flax schedulers (#508 ) * beta never changes removed from state * fix typos in docs * removed unused var * initial ddim flax scheduler * import * added dummy objects * fix style * fix typo * docs * fix typo in comment * set return type * added flax ddom * fix style * remake * pass PRNG key as argument and split before use * fix doc string * use config * added flax Karras VE scheduler * make style * fix dummy * fix ndarray type annotation * replace returns a new state * added lms_discrete scheduler * use self.config * add_noise needs state * use config * use config * docstring * added flax score sde ve * fix imports * fix typos	2022-09-15 15:55:48 +02:00
Patrick von Platen	cc59b05635	[ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334 ) * add outputs for models * add for pipelines * finish schedulers * better naming * adapt tests as well * replace dict access with . access * make schedulers works * finish * correct readme * make bcp compatible * up * small fix * finish * more fixes * more fixes * Apply suggestions from code review Co-authored-by: Suraj Patil <surajp815@gmail.com> Co-authored-by: Pedro Cuenca <pedro@huggingface.co> * Update src/diffusers/models/vae.py Co-authored-by: Pedro Cuenca <pedro@huggingface.co> * Adapt model outputs * Apply more suggestions * finish examples * correct Co-authored-by: Suraj Patil <surajp815@gmail.com> Co-authored-by: Pedro Cuenca <pedro@huggingface.co>	2022-09-05 14:49:26 +02:00
Anton Lozhkov	4724250980	Fix nondeterministic tests for GPU runs (#314 ) * Fix nondeterministic tests for GPU runs * force SD fast tests to the CPU	2022-09-01 15:25:39 +02:00
Anton Lozhkov	6c15636b0b	Add training and batched inference test for DDPM vs DDIM (#140 ) * Add torch_device to the VE pipeline * Mark the training test with slow	2022-07-27 15:01:56 +02:00

4 Commits