mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-02-19 20:04:04 +00:00
Initial support for finetuning qwen image. Will only work with caching for now, need to add controls everywhere.
This commit is contained in:
@@ -50,6 +50,7 @@ class FileItemDTO(
|
||||
self.is_video = self.dataset_config.num_frames > 1
|
||||
size_database = kwargs.get('size_database', {})
|
||||
dataset_root = kwargs.get('dataset_root', None)
|
||||
self.encode_control_in_text_embeddings = kwargs.get('encode_control_in_text_embeddings', False)
|
||||
if dataset_root is not None:
|
||||
# remove dataset root from path
|
||||
file_key = self.path.replace(dataset_root, '')
|
||||
|
||||
Reference in New Issue
Block a user