File size: 617 Bytes
dc76781 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | {
"accum_steps": 1,
"architectures": [
"FWKVVisionModel"
],
"batch_size": 32,
"clip_id": "openai/clip-vit-base-patch32",
"data_source": "t2i2m",
"depth": 12,
"dtype": "float32",
"epochs": 5,
"ffn_mult": 4,
"image_size": 256,
"latent_cache_path": "fwkv_vision_latent_cache.pt",
"latent_ch": 4,
"latent_size": 32,
"lr": 0.0001,
"model_type": "fwkv_vision",
"n_examples": 100000,
"n_heads": 6,
"patch": 2,
"save_dir": "fwkv_vision_ckpt",
"text_dim": 512,
"transformers_version": "5.0.0",
"vae_id": "stabilityai/sd-vae-ft-mse",
"vae_scaling": 0.18215,
"width": 384
}
|