Upload 4 files

maskgit.6845000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf8f95ffbf410d548a4eed67ff7af4506a0507a5c99c901419eda33456a0d56d
+size 1918653415

maskgit.6845000.pt.yaml

@@ -0,0 +1,78 @@
+total_params: 345393868
+image_size: 256
+num_tokens: 8192
+num_train_steps: -1
+num_epochs: 50
+dim: 32
+channels: 3
+batch_size: 1
+lr: 0.0001
+gradient_accumulation_steps: 100
+save_results_every: 500
+save_model_every: 5000
+vq_codebook_size: 8192
+vq_codebook_dim: 8192
+lr_scheduler: cosine_with_restarts
+lr_warmup_steps: 10
+seq_len: 1024
+depth: 4
+dim_head: 64
+heads: 8
+ff_mult: 4
+t5_name: t5-large
+mixed_precision: 'no'
+cond_image_size: null
+validation_prompt: a girl in a red dress|a black cat|a dog with white hair and sun
+  glasses
+timesteps: 18
+optimizer: Adam
+only_save_last_checkpoint: false
+validation_image_scale: 1.0
+no_center_crop: false
+no_flip: false
+dataset_save_path: E:\cached_datasets\INE2
+clear_previous_experiments: false
+max_grad_norm: null
+seed: 42
+valid_frac: 0.05
+use_ema: false
+ema_beta: 0.995
+ema_update_after_step: 1
+ema_update_every: 1
+apply_grad_penalty_every: 4
+image_column: image
+caption_column: caption
+log_with: wandb
+use_8bit_adam: false
+results_dir: results\Muse-v0.9_test
+logging_dir: null
+vae_path: results\Muse-v0.9_test\vae.13680000.pt
+dataset_name: null
+hf_split_name: null
+streaming: false
+train_data_dir: E:\dataset
+checkpoint_limit: null
+cond_drop_prob: 0.5
+scheduler_power: 16.0
+num_cycles: 800
+resume_path: results\Muse-v0.9_test\maskgit.6775000.pt
+taming_model_path: null
+taming_config_path: null
+weight_decay: 0.0001
+cache_path: null
+no_cache: true
+link: false
+latest_checkpoint: true
+do_not_save_config: false
+use_l2_recon_loss: false
+debug: false
+config_path: null
+attention_type: ein
+precompute: false
+precompute_path: ''
+random_crop: true
+project_name: muse_maskgit
+run_name: null
+wandb_user: sygil
+layers: 4
+discr_layers: 4

vae.13740000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e55b500576e40148b4127c687b3bac3c23af0184599de899613ab9636cc5c176
+size 577198907

vae.13740000.pt.yaml

@@ -0,0 +1,67 @@
+total_params: 10067428
+image_size: 256
+num_train_steps: -1
+num_epochs: 50
+batch_size: 5
+lr: 0.0001
+lr_warmup_steps: 10
+lr_scheduler: cosine_with_restarts
+gradient_accumulation_steps: 2000
+save_results_every: 500
+save_model_every: 5000
+dim: 32
+vq_codebook_size: 8192
+vq_codebook_dim: 8192
+seq_len: 1024
+channels: 3
+layers: 4
+discr_layers: 4
+scheduler_power: 8.0
+num_cycles: 800
+only_save_last_checkpoint: false
+validation_image_scale: 1.0
+no_center_crop: false
+no_flip: false
+random_crop: true
+dataset_save_path: E:/cached_datasets/INE
+clear_previous_experiments: false
+max_grad_norm: null
+discr_max_grad_norm: null
+seed: 42
+valid_frac: 0.05
+use_ema: false
+ema_beta: 0.995
+ema_update_after_step: 1
+ema_update_every: 1
+apply_grad_penalty_every: 4
+image_column: image
+caption_column: caption
+log_with: wandb
+mixed_precision: 'no'
+use_8bit_adam: false
+results_dir: results\Muse-v0.9_test
+logging_dir: null
+resume_path: results\Muse-v0.9_test\vae.13385000.pt
+dataset_name: null
+streaming: false
+train_data_dir: F:/Hydrus Files
+checkpoint_limit: null
+cond_drop_prob: 0.5
+taming_model_path: null
+taming_config_path: null
+optimizer: Adam
+weight_decay: 0.001
+cache_path: null
+no_cache: true
+latest_checkpoint: true
+do_not_save_config: false
+use_l2_recon_loss: false
+debug: false
+config_path: null
+validation_folder_at_end_of_epoch: samples/validation
+exclude_folders: null
+webdataset: null
+project_name: muse_vae
+run_name: null
+wandb_user: null
+hf_split_name: train