checkpoint-300/config.json

{
  "architectures": [
    "ImgGen_HF"
  ],
  "bnb_4bit_quant_type": "nf4",
  "cls_token_num": 120,
  "codebook_embed_dim": 8,
  "codebook_size": 16384,
  "downsample_size": 16,
  "drop_path_rate": 0.0,
  "dropout_p": 0.0,
  "gpt_model": "GPT-XL",
  "gpt_type": "t2i",
  "image_size": 256,
  "load_in_4bit": false,
  "load_in_8bit": false,
  "lora_alpha": 32,
  "lora_dropout": 0.05,
  "lora_modules_to_save": null,
  "lora_r": 16,
  "lora_target_modules": null,
  "lora_task_type": "CAUSAL_LM",
  "model_name_or_path": "custom/noneexists",
  "model_revision": "main",
  "model_type": "ImgGen_HF",
  "text_encoder": "google/flan-t5-xl",
  "text_encoder_ckpt": "",
  "text_feature_max_len": 120,
  "token_dropout_p": 0.0,
  "torch_dtype": "bfloat16",
  "transformers_version": "4.53.0",
  "trust_remote_code": false,
  "use_bnb_nested_quant": false,
  "use_dora": false,
  "use_peft": false,
  "use_rslora": false,
  "vq_model": "VQ-16"
}