Upload folder using huggingface_hub

checkpoint-21600-lr/README.md

@@ -0,0 +1,202 @@
+---
+library_name: peft
+base_model: segmind/Segmind-Vega
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+
+
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+[More Information Needed]
+
+### Downstream Use [optional]
+
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
+
+[More Information Needed]
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+[More Information Needed]
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+[More Information Needed]
+
+### Recommendations
+
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
+
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+[More Information Needed]
+
+### Training Procedure
+
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
+
+#### Preprocessing [optional]
+
+[More Information Needed]
+
+
+#### Training Hyperparameters
+
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]
+### Framework versions
+
+- PEFT 0.9.0

checkpoint-21600-lr/adapter_config.json

@@ -0,0 +1,43 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": {
+    "base_model_class": "UNet2DConditionModel",
+    "parent_library": "diffusers.models.unets.unet_2d_condition"
+  },
+  "base_model_name_or_path": null,
+  "bias": "none",
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 8,
+  "lora_dropout": 0.0,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 64,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "proj_in",
+    "proj_out",
+    "conv_shortcut",
+    "conv1",
+    "to_v",
+    "to_q",
+    "downsamplers.0.conv",
+    "ff.net.0.proj",
+    "to_out.0",
+    "conv2",
+    "upsamplers.0.conv",
+    "ff.net.2",
+    "time_emb_proj",
+    "to_k"
+  ],
+  "task_type": null,
+  "use_dora": false,
+  "use_rslora": false
+}

checkpoint-21600-lr/adapter_model.safetensors

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

checkpoint-21600-lr/optimizer.bin

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

checkpoint-21600-lr/optimizer_1.bin

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

checkpoint-21600-lr/random_states_0.pkl

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

checkpoint-21600-lr/scaler.pt

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

checkpoint-21600-lr/scheduler.bin

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

checkpoint-21600-lr/unet_lora/pytorch_lora_weights.safetensors

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

checkpoint-21600/heads.pth

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

checkpoint-21600/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2fe453fd0776abef0913967b89b3f5e9b00aaaa1411d6f7f20549f95adc4d43a
-size 121536062
+oid sha256:15a1b9798208ed29add78db82eb5aad35767886e3f10075e34f4a6ff3c769609
+size 1496291698

checkpoint-21600/optimizer_1.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac1e8b3cea6ea21b80a567d48d8a4caf7f08a643c8a0ed6467c832786a7724de
+oid sha256:f693c7b62f5d97fb7d846038ce06db80b608432888fff512cf2f155c16e2abfa
 size 15431330

checkpoint-21600/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0b42437eed92eda25f3e88b90efa94157e6abd838159058295b540e399ad251e
+oid sha256:4c93274b9864ff62e853f4c5a43ffd0f8d0d9151f37d654ddc8732abae897889
 size 14344

checkpoint-21600/scaler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:899a642947e70d7e60cf2eeee96b1e912dfb435fbf6e019478dc646f870906c8
+oid sha256:65a89af3e28ba524872d3298f7b6cf7bec8fec6408b688a24360f195c43eda10
 size 988

checkpoint-21600/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:695c9a54d114da7c87ef5d6deb506bd1a902ca35b162e5e6564d711d9593f60a
+oid sha256:419509e6541bd5d5e8be2235aed61c5578c0804d41e1f56a0397dde69c4ff8f6
 size 1000

checkpoint-21600/unet/config.json

@@ -0,0 +1,73 @@
+{
+  "_class_name": "UNet2DConditionModel",
+  "_diffusers_version": "0.26.3",
+  "_name_or_path": "k5pcmlra2-test",
+  "act_fn": "silu",
+  "addition_embed_type": "text_time",
+  "addition_embed_type_num_heads": 64,
+  "addition_time_embed_dim": 256,
+  "attention_head_dim": [
+    5,
+    10,
+    20
+  ],
+  "attention_type": "default",
+  "block_out_channels": [
+    320,
+    640,
+    1280
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "class_embeddings_concat": false,
+  "conv_in_kernel": 3,
+  "conv_out_kernel": 3,
+  "cross_attention_dim": 2048,
+  "cross_attention_norm": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "dropout": 0.0,
+  "dual_cross_attention": false,
+  "encoder_hid_dim": null,
+  "encoder_hid_dim_type": null,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "mid_block_only_cross_attention": null,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2D",
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_attention_heads": null,
+  "num_class_embeds": null,
+  "only_cross_attention": false,
+  "out_channels": 4,
+  "projection_class_embeddings_input_dim": 2816,
+  "resnet_out_scale_factor": 1.0,
+  "resnet_skip_time_act": false,
+  "resnet_time_scale_shift": "default",
+  "reverse_transformer_layers_per_block": null,
+  "sample_size": 128,
+  "time_cond_proj_dim": null,
+  "time_embedding_act_fn": null,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "timestep_post_act": null,
+  "transformer_layers_per_block": [
+    1,
+    1,
+    2
+  ],
+  "up_block_types": [
+    "CrossAttnUpBlock2D",
+    "CrossAttnUpBlock2D",
+    "UpBlock2D"
+  ],
+  "upcast_attention": null,
+  "use_linear_projection": true
+}

checkpoint-21600/unet/diffusion_pytorch_model.safetensors

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

logs/text2image-fine-tune/1758023363.4801242/events.out.tfevents.1758023363.1edfd6e8f17a.4024.1

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

logs/text2image-fine-tune/1758023363.4925895/hparams.yml

@@ -0,0 +1,55 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0
+adv_lr: 1.0e-05
+adv_weight: 0.1
+allow_tf32: true
+cache_dir: null
+cast_teacher_unet: false
+center_crop: false
+checkpointing_steps: 200
+checkpoints_total_limit: 10
+dataloader_num_workers: 1
+enable_xformers_memory_efficient_attention: true
+gradient_accumulation_steps: 10
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+huber_c: 0.001
+learning_rate: 2.0e-06
+local_rank: -1
+logging_dir: logs
+lora_rank: 64
+loss_type: huber
+lr_scheduler: constant
+lr_warmup_steps: 500
+max_grad_norm: 1.0
+max_train_samples: 4000000
+max_train_steps: 21600
+mixed_precision: fp16
+multiphase: 1
+not_apply_cfg_solver: false
+num_ddim_timesteps: 40
+num_train_epochs: 108
+output_dir: k5pcmlra2-test
+pretrained_teacher_model: segmind/Segmind-Vega
+pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
+proportion_empty_prompts: 0
+push_to_hub: false
+random_flip: false
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: checkpoint-21400
+revision: null
+scale_lr: false
+seed: null
+teacher_revision: null
+tracker_project_name: text2image-fine-tune
+train_batch_size: 4
+train_shards_path_or_url: null
+use_8bit_adam: true
+use_fix_crop_and_size: false
+validation_steps: 500000
+w_max: 7.0
+w_min: 6.0

logs/text2image-fine-tune/events.out.tfevents.1758023363.1edfd6e8f17a.4024.0

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