| { | |
| "_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 | |
| } | |
| } | |