I have tried to run your code, the problem is i am getting this error, i am using the same dataset i used with the ai-toolkit tool:
Starting 1 job(s)
Starting job train_lora_sdxl_24gb_1.0.yaml
Using device: cuda:0
Using TF32 for faster training on Ampere GPUs
Using matmul precision: high
Using cuDNN benchmark
Using seed: 42
Using variant: fp16
Using 4 workers for data loading
Walking seed
Loading noise scheduler
{'variance_type', 'clip_sample_range', 'thresholding', 'rescale_betas_zero_snr', 'dynamic_thresholding_ratio'} was not found in config. Values will be initialized to default values.
Using DDPMScheduler
Loading text encoder 1
You are using a model of type clip_text_model to instantiate a model of type . This is not supported for all configurations of models and can yield errors.
Loading text encoder 2
You are using a model of type clip_text_model to instantiate a model of type . This is not supported for all configurations of models and can yield errors.
Loading VAE
{'latents_mean', 'use_quant_conv', 'use_post_quant_conv', 'mid_block_add_attention', 'shift_factor', 'latents_std'} was not found in config. Values will be initialized to default values.
Loading unet
{'dropout', 'attention_type', 'reverse_transformer_layers_per_block'} was not found in config. Values will be initialized to default values.
Adding LoRA adapters
Loading optimizer
Loading datasets
Job train_lora_sdxl_24gb_1.0.yaml failed with error: Instance images root doesn't exists.
Traceback (most recent call last):
File "/workspace/lorakit/.venv/lib/python3.10/site-packages/lorakit/cli/lorakit.py", line 31, in main
job.run()
File "/workspace/lorakit/.venv/lib/python3.10/site-packages/lorakit/train.py", line 762, in run
train_dataset = DreamBoothDataset(self._dataset_folder, self._instant_prompt, self._class_prompt, resolution=self._resolution)
File "/workspace/lorakit/.venv/lib/python3.10/site-packages/lorakit/datasets.py", line 41, in init
raise ValueError("Instance images root doesn't exists.")
ValueError: Instance images root doesn't exists.
I have tried to run your code, the problem is i am getting this error, i am using the same dataset i used with the ai-toolkit tool:
Starting 1 job(s)
Starting job train_lora_sdxl_24gb_1.0.yaml
Using device: cuda:0
Using TF32 for faster training on Ampere GPUs
Using matmul precision: high
Using cuDNN benchmark
Using seed: 42
Using variant: fp16
Using 4 workers for data loading
Walking seed
Loading noise scheduler
{'variance_type', 'clip_sample_range', 'thresholding', 'rescale_betas_zero_snr', 'dynamic_thresholding_ratio'} was not found in config. Values will be initialized to default values.
Using DDPMScheduler
Loading text encoder 1
You are using a model of type clip_text_model to instantiate a model of type . This is not supported for all configurations of models and can yield errors.
Loading text encoder 2
You are using a model of type clip_text_model to instantiate a model of type . This is not supported for all configurations of models and can yield errors.
Loading VAE
{'latents_mean', 'use_quant_conv', 'use_post_quant_conv', 'mid_block_add_attention', 'shift_factor', 'latents_std'} was not found in config. Values will be initialized to default values.
Loading unet
{'dropout', 'attention_type', 'reverse_transformer_layers_per_block'} was not found in config. Values will be initialized to default values.
Adding LoRA adapters
Loading optimizer
Loading datasets
Job train_lora_sdxl_24gb_1.0.yaml failed with error: Instance images root doesn't exists.
Traceback (most recent call last):
File "/workspace/lorakit/.venv/lib/python3.10/site-packages/lorakit/cli/lorakit.py", line 31, in main
job.run()
File "/workspace/lorakit/.venv/lib/python3.10/site-packages/lorakit/train.py", line 762, in run
train_dataset = DreamBoothDataset(self._dataset_folder, self._instant_prompt, self._class_prompt, resolution=self._resolution)
File "/workspace/lorakit/.venv/lib/python3.10/site-packages/lorakit/datasets.py", line 41, in init
raise ValueError("Instance images root doesn't exists.")
ValueError: Instance images root doesn't exists.