File size: 1,835 Bytes
1e96800 |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
pretrained_model_path: ./models/zeroscope_v2_576w/
output_dir: ./outputs
train_data:
width: 1024
height: 576
use_bucketing: true
sample_start_idx: 1
fps: 24
frame_step: 1
n_sample_frames: 1
single_video_path: path/to/single/video.mp4
single_video_prompt: ''
fallback_prompt: ''
path: path/to/folder/of/videos/
json_path: C:\potat1_dataset\potat1_blip2.json
image_dir: path/to/image/directory
single_img_prompt: ''
validation_data:
prompt: ''
sample_preview: false
num_frames: 16
width: 1024
height: 576
num_inference_steps: 25
guidance_scale: 9
extra_train_data: []
dataset_types:
- json
shuffle: true
validation_steps: 10000
extra_unet_params: null
extra_text_encoder_params: null
train_batch_size: 1
max_train_steps: 50000
learning_rate: 5.0e-06
scale_lr: false
lr_scheduler: constant
lr_warmup_steps: 0
adam_beta1: 0.9
adam_beta2: 0.999
adam_weight_decay: 0.01
adam_epsilon: 1.0e-08
max_grad_norm: 1.0
gradient_accumulation_steps: 1
gradient_checkpointing: true
text_encoder_gradient_checkpointing: false
checkpointing_steps: 10000
resume_from_checkpoint: null
resume_step: null
mixed_precision: fp16
use_8bit_adam: false
enable_xformers_memory_efficient_attention: true
enable_torch_2_attn: true
seed: 64
extend_dataset: false
cached_latent_dir: null
lora_version: stable_lora
save_lora_for_webui: true
only_lora_for_webui: false
lora_bias: none
use_unet_lora: false
use_text_lora: false
unet_lora_modules:
- UNet3DConditionModel
text_encoder_lora_modules:
- CLIPEncoderLayer
save_pretrained_model: true
lora_rank: 16
lora_path: ''
lora_unet_dropout: 0.1
lora_text_dropout: 0.1
logger_type: tensorboard
cache_latents: true
kwargs: {}
offset_noise_strength: 0.1
rescale_schedule: false
train_text_encoder: true
trainable_modules:
- all
trainable_text_modules:
- all
use_offset_noise: false
|