Hugging Face's logo

BlueSourceJY
/
SiT-Complementary

Model card Files
xet
 Community
SiT-Complementary
File size: 2,384 Bytes
8b0b874
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
{
  "base": {
    "source_path": "/home/nvidia/SiT-official-bs256/results-official-bs256-400k/000-SiT-S-2-Linear-velocity-None/checkpoints/4003200.pt",
    "filename": "4003200.pt",
    "bytes": 526844432,
    "sha256": "81c9246463c94a3200e7d1861a95bd923e3f3cac92e4966d861a843348142f4e",
    "metadata_fields": [
      "args",
      "ema",
      "model",
      "opt"
    ],
    "checkpoint_steps": null,
    "configured_epochs": 800,
    "global_batch_size": 256,
    "learning_rate": null,
    "model_module": "models",
    "state_dict": "ema",
    "learn_sigma": true,
    "strict_load": true,
    "forward_shape": [
      1,
      4,
      32,
      32
    ],
    "decoded_shape": [
      1,
      3,
      256,
      256
    ],
    "preview": "base_ema_euler10_seed20260815.png"
  },
  "rot-layer": {
    "source_path": "/home/nvidia/SiT-rot-layer-bs256/results-200ep/SiT-S-2-RotLayer-bs256-lr1e-4-200ep/checkpoints/4003200.pt",
    "filename": "4003200.pt",
    "bytes": 526845456,
    "sha256": "0d0a191eda63f82d6262f8b41644a3f9ceb0f3cfe66cc0accc711c8b488bbc32",
    "metadata_fields": [
      "args",
      "ema",
      "model",
      "opt"
    ],
    "checkpoint_steps": null,
    "configured_epochs": 800,
    "global_batch_size": 256,
    "learning_rate": 0.0001,
    "model_module": "models_rot_layer",
    "state_dict": "ema",
    "learn_sigma": true,
    "strict_load": true,
    "forward_shape": [
      1,
      4,
      32,
      32
    ],
    "decoded_shape": [
      1,
      3,
      256,
      256
    ],
    "preview": "rot-layer_ema_euler10_seed20260815.png"
  },
  "conv-layer": {
    "source_path": "/data/nvidia/SiT-conv-layer-bs256/results-800ep/SiT-S-2-ConvLayer-bs256-lr1e-4-800ep/checkpoints/1950000.pt",
    "filename": "1950000.pt",
    "bytes": 526985776,
    "sha256": "f3724fa2651c6fbb3a624664f057c1dd56c658d68fb356c52cf51a7684ae7548",
    "metadata_fields": [
      "args",
      "ema",
      "model",
      "opt"
    ],
    "checkpoint_steps": null,
    "configured_epochs": 800,
    "global_batch_size": 256,
    "learning_rate": 0.0001,
    "model_module": "models_conv",
    "state_dict": "ema",
    "learn_sigma": true,
    "strict_load": true,
    "forward_shape": [
      1,
      4,
      32,
      32
    ],
    "decoded_shape": [
      1,
      3,
      256,
      256
    ],
    "preview": "conv-layer_ema_euler10_seed20260815.png"
  }
}