This is an inference-focused library initially for simplicity. We released a full model definition including our causal VAE and context parallel inference so you could try to implement a tuning loop. It's a bit computationally intensive to pre-train at the moment.
Hi Yachty,
This is an inference-focused library initially for simplicity. We released a full model definition including our causal VAE and context parallel inference so you could try to implement a tuning loop. It's a bit computationally intensive to pre-train at the moment.