Hugging Face's logo

Spaces:
tencent
/
Hunyuan3D-2.1
Running on Zero

App Files Community
21
Hunyuan3D-2.1
File size: 1,240 Bytes 
600759a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
model:
  base_learning_rate: 5.0e-05
  target: hunyuanpaintpbr.model.HunyuanPaint
  params:
    num_view: 6
    view_size: 512
    drop_cond_prob: 0.1
    
    noise_in_channels: 12

    stable_diffusion_config:
      pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
      custom_pipeline: ./hunyuanpaintpbr
      

data:
  target: src.data.objaverse_hunyuan.DataModuleFromConfig
  params:
    batch_size: 1
    num_workers: 4
    train:
      - 
        target: src.data.dataloader.objaverse_loader_forTexturePBR.TextureDataset
        params:
          num_view: 6
          json_path: train_examples/examples.json
    validation:
      -
        target: src.data.dataloader.objaverse_loader_forTexturePBR.TextureDataset
        params:
          num_view: 6
          json_path: train_examples/examples.json

lightning:
  modelcheckpoint:
    params:
      every_n_train_steps: 10000
      save_top_k: -1
      save_last: true
  callbacks: {}

  trainer:
    benchmark: true
    max_epochs: -1
    gradient_clip_val: 1.0
    val_check_interval: 1000
    num_sanity_val_steps: 0
    accumulate_grad_batches: 1
    check_val_every_n_epoch: null   # if not set this, validation does not run

init_control_from:
resume_from: