diffusion

Scalable Diffusion Models with Transformers

Classes

Diffusion([predict_sequence_length, config])	TensorFlow Diffusion model for time series forecasting
DiffusionConfig([hidden_size, num_layers, ...])	Initializes the configuration for the Diffusion model with the specified parameters.
NoiseScheduler(config)	Linear noise scheduler for diffusion models
TransformerBlock(*args, **kwargs)	Transformer block for Diffusion model