config.json

{
    "_class_name": "DiffusionPipeline",  // Replace with the appropriate class name for your model
    "model_type": "Image-Generation",      // e.g., "unet", "autoencoder", etc.
    "params": {
        "num_train_timesteps": 1000,      // Number of timesteps for training
        "image_size": 256,                 // Size of the images your model expects
        "channels": 3,                     // Number of input channels (3 for RGB)
        "use_fp16": true,                  // Use float16 precision (true/false)
        "dropout": 0.1                     // Dropout rate, if applicable
    },
    "scheduler": {
        "type": "pndm",                    // Scheduler type (e.g., "pndm", "ddim", etc.)
        "beta_start": 0.00085,             // Start value for beta
        "beta_end": 0.012,                 // End value for beta
        "beta_schedule": "linear"           // Schedule for beta updates
    },
    "tokenizer": {
        "type": "simple_tokenizer",         // Type of tokenizer used
        "vocab_size": 30522                 // Size of the vocabulary
    }
}