Hugging Face's logo

Spaces:
Himanshu806
/
Leffa
Running on Zero

App Files Community
Leffa
File size: 2,312 Bytes 
b213d84
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
viton_hd_train:
  dataset:
    _target_: media_dataloader.api.EnrichingDataset
    datasource:
      _target_: media_dataloader.api.LazyHiveDataSource
      namespace: ad_metrics
      table: vton_public_dataset_viton_hd_train_v2
      partition_filter_predicate_list: ["ds = '2024-10-30'"]
    enrichments:
      - _target_: media_dataloader.api.media_lookups.ManifoldLookups
        lookup_handle_to_media_columns:
          image_manifold_path: "image"
          cloth_manifold_path: "cloth"
          agnostic_mask_manifold_path: "agnostic_mask"
          image_densepose_manifold_path: "image_densepose"
          cloth_mask_manifold_path: "cloth_mask"
          image_parse_v3_manifold_path: "image_parse"
    collate_fn:
      - _target_: media_dataloader.api.Collate
      - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
        image_field: image
        blob_field: image
      - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
        image_field: cloth
        blob_field: cloth
      - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
        image_field: agnostic_mask
        blob_field: agnostic_mask
      - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
        image_field: image_densepose
        blob_field: image_densepose
      - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
        image_field: cloth_mask
        blob_field: cloth_mask
      - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform
        image_field: image_parse
        blob_field: image_parse
      - _target_: leffa.datasets.transform.VtonTransform
        height: ${constants.height}
        width: ${constants.width}
        is_train: true
        dataset: viton_hd
        aug_garment_ratio: 0.0
        get_garment_from_person_ratio: 0.0
        aug_mask_ratio: 0.0

  dataloader:
    _target_: media_dataloader.api.StatefulDataLoader
    dataset: ${datasets.viton_hd_train.dataset}
    batch_size: ${constants.batch_size}
    num_workers: 4
    prefetch_factor: 2
    pin_memory: true
    persistent_workers: true
    multiprocessing_context: forkserver