Commit ·
4da163f
1
Parent(s): 6abe861
made some corrections
Browse files- __pycache__/blender_backend.cpython-310.pyc +0 -0
- __pycache__/infer_backend.cpython-310.pyc +0 -0
- __pycache__/object_scales.cpython-310.pyc +0 -0
- train/src/__pycache__/jsonl_datasets.cpython-310.pyc +0 -0
- train/src/__pycache__/transformer_flux.cpython-310.pyc +0 -0
- train/train.py +0 -2
- train/train.sh +2 -3
__pycache__/blender_backend.cpython-310.pyc
ADDED
|
Binary file (31.9 kB). View file
|
|
|
__pycache__/infer_backend.cpython-310.pyc
ADDED
|
Binary file (9.42 kB). View file
|
|
|
__pycache__/object_scales.cpython-310.pyc
ADDED
|
Binary file (7.53 kB). View file
|
|
|
train/src/__pycache__/jsonl_datasets.cpython-310.pyc
CHANGED
|
Binary files a/train/src/__pycache__/jsonl_datasets.cpython-310.pyc and b/train/src/__pycache__/jsonl_datasets.cpython-310.pyc differ
|
|
|
train/src/__pycache__/transformer_flux.cpython-310.pyc
CHANGED
|
Binary files a/train/src/__pycache__/transformer_flux.cpython-310.pyc and b/train/src/__pycache__/transformer_flux.cpython-310.pyc differ
|
|
|
train/train.py
CHANGED
|
@@ -1098,8 +1098,6 @@ def main(args):
|
|
| 1098 |
enumerated_dataloader = enumerate(dataloader_iterator, start=skip_steps)
|
| 1099 |
else:
|
| 1100 |
enumerated_dataloader = enumerate(train_dataloader)
|
| 1101 |
-
if epoch == first_epoch:
|
| 1102 |
-
continue
|
| 1103 |
for step, batch in enumerated_dataloader:
|
| 1104 |
progress_bar.set_description(f"epoch {epoch}, dataset_ids: {batch['index']}")
|
| 1105 |
models_to_accumulate = [transformer]
|
|
|
|
| 1098 |
enumerated_dataloader = enumerate(dataloader_iterator, start=skip_steps)
|
| 1099 |
else:
|
| 1100 |
enumerated_dataloader = enumerate(train_dataloader)
|
|
|
|
|
|
|
| 1101 |
for step, batch in enumerated_dataloader:
|
| 1102 |
progress_bar.set_description(f"epoch {epoch}, dataset_ids: {batch['index']}")
|
| 1103 |
models_to_accumulate = [transformer]
|
train/train.sh
CHANGED
|
@@ -1,9 +1,8 @@
|
|
| 1 |
export MODEL_DIR="black-forest-labs/FLUX.1-dev" # your flux path
|
| 2 |
export OUTPUT_DIR="/archive/vaibhav.agrawal/a-bev-of-the-latents/easycontrol_cuboids" # your save path
|
| 3 |
export CONFIG="./default_config.yaml"
|
| 4 |
-
export TRAIN_DATA="/archive/vaibhav.agrawal/a-bev-of-the-latents/
|
| 5 |
export LOG_PATH="$OUTPUT_DIR/log"
|
| 6 |
-
export INFERENCE_EMBEDS_DIR="/archive/vaibhav.agrawal/a-bev-of-the-latents/inference_embeds_datasetv7_superhard"
|
| 7 |
|
| 8 |
accelerate launch --config_file $CONFIG train.py \
|
| 9 |
--pretrained_model_name_or_path $MODEL_DIR \
|
|
@@ -24,4 +23,4 @@ accelerate launch --config_file $CONFIG train.py \
|
|
| 24 |
--train_batch_size=1 \
|
| 25 |
--stage1_epochs=1 \
|
| 26 |
--stage2_steps=5000 \
|
| 27 |
-
--checkpointing_steps=
|
|
|
|
| 1 |
export MODEL_DIR="black-forest-labs/FLUX.1-dev" # your flux path
|
| 2 |
export OUTPUT_DIR="/archive/vaibhav.agrawal/a-bev-of-the-latents/easycontrol_cuboids" # your save path
|
| 3 |
export CONFIG="./default_config.yaml"
|
| 4 |
+
export TRAIN_DATA="/archive/vaibhav.agrawal/a-bev-of-the-latents/datasetv9/rgb.jsonl" # your data jsonl file
|
| 5 |
export LOG_PATH="$OUTPUT_DIR/log"
|
|
|
|
| 6 |
|
| 7 |
accelerate launch --config_file $CONFIG train.py \
|
| 8 |
--pretrained_model_name_or_path $MODEL_DIR \
|
|
|
|
| 23 |
--train_batch_size=1 \
|
| 24 |
--stage1_epochs=1 \
|
| 25 |
--stage2_steps=5000 \
|
| 26 |
+
--checkpointing_steps=5000
|