Hugging Face's logo

SciLM
/
ResFlow

Safetensors
resflow
resflow-dit3d
geology
reservoir
subsurface
3d
flow-matching
diffusion-transformer
generative
Model card Files
xet
 Community
ResFlow / config.json
baghishov's picture
baghishov
Sample with the paper's Euler 100 steps; model card with the paper's final results.
58f12de verified
Raw History Blame Contribute Delete
5.07 kB
{
"model_type": "resflow-dit3d",
"architecture": {
"in_channels": 3,
"out_channels": 1,
"volume_shape": [
64,
64,
32
],
"patch_size": [
4,
4,
2
],
"hidden": 384,
"depth": 12,
"num_heads": 6,
"num_cond": 18,
"num_time_embs": 1,
"expand_angle_idx": null,
"qk_norm": true,
"conv_io": false,
"pos_embed": "rope",
"rope_theta": 100.0,
"window": null,
"window_shift": false,
"token_conv": 0,
"token_conv_kernel": [
3,
3,
3
],
"attn_mode": "window",
"attn_radius": [
8,
8,
8
]
},
"layer_types": [
"lobe",
"channel:PV_SHOESTRING",
"channel:CB_LABYRINTH",
"channel:CB_JIGSAW",
"channel:SH_DISTAL",
"channel:SH_PROXIMAL",
"channel:MEANDER_OXBOW",
"delta"
],
"cont_cols": [
"ntg",
"width_cells",
"depth_cells",
"asp",
"mCHsinu",
"mFFCHprop",
"probAvulInside",
"trunk_length_fraction"
],
"n_universal": 3,
"cont_min": [
0.05118560791015625,
2.404618263244629,
3.000000238418579,
1.000009536743164,
1.0500000715255737,
0.0,
0.0,
0.10000130534172058
],
"cont_max": [
0.9156951904296875,
79.82032775878906,
19.993122100830078,
2.499992847442627,
1.8499996662139893,
0.6999967098236084,
0.7999995946884155,
0.4999952018260956
],
"env_defaults": {
"lobe": {
"ntg": 0.517,
"width_cells": 46.0001,
"depth_cells": 12.4999,
"asp": 1.75
},
"channel:PV_SHOESTRING": {
"ntg": 0.2077,
"width_cells": 7.5233,
"depth_cells": 6.9282,
"mCHsinu": 1.575,
"mFFCHprop": 0.15,
"probAvulInside": 0.06
},
"channel:CB_LABYRINTH": {
"ntg": 0.357,
"width_cells": 10.9679,
"depth_cells": 6.9282,
"mCHsinu": 1.275,
"mFFCHprop": 0.325,
"probAvulInside": 0.06
},
"channel:CB_JIGSAW": {
"ntg": 0.388,
"width_cells": 11.6044,
"depth_cells": 6.9282,
"mCHsinu": 1.25,
"mFFCHprop": 0.4,
"probAvulInside": 0.375
},
"channel:SH_DISTAL": {
"ntg": 0.6356,
"width_cells": 12.3673,
"depth_cells": 6.9282,
"mCHsinu": 1.275,
"mFFCHprop": 0.1,
"probAvulInside": 0.06
},
"channel:SH_PROXIMAL": {
"ntg": 0.5099,
"width_cells": 14.4069,
"depth_cells": 6.9282,
"mCHsinu": 1.225,
"mFFCHprop": 0.225,
"probAvulInside": 0.35
},
"channel:MEANDER_OXBOW": {
"ntg": 0.494,
"width_cells": 7.5244,
"depth_cells": 6.9282,
"mCHsinu": 1.625,
"mFFCHprop": 0.0,
"probAvulInside": 0.0
},
"delta": {
"ntg": 0.1934,
"width_cells": 10.1923,
"depth_cells": 6.9282,
"mCHsinu": 1.075,
"mFFCHprop": 0.2,
"probAvulInside": 0.65,
"trunk_length_fraction": 0.3
}
},
"env_ranges": {
"lobe": {
"ntg": [
0.0448,
0.9392
],
"width_cells": [
12.1034,
79.9134
],
"depth_cells": [
5.0161,
19.9931
],
"asp": [
1.0,
2.5
]
},
"channel:PV_SHOESTRING": {
"ntg": [
0.0,
0.6591
],
"width_cells": [
2.4042,
22.3059
],
"depth_cells": [
3.0,
15.9999
],
"mCHsinu": [
1.3,
1.85
],
"mFFCHprop": [
0.0,
0.3
],
"probAvulInside": [
0.02,
0.1
]
},
"channel:CB_LABYRINTH": {
"ntg": [
0.0,
0.9188
],
"width_cells": [
3.6057,
31.8716
],
"depth_cells": [
3.0,
15.9999
],
"mCHsinu": [
1.1,
1.45
],
"mFFCHprop": [
0.1,
0.55
],
"probAvulInside": [
0.02,
0.1
]
},
"channel:CB_JIGSAW": {
"ntg": [
0.0,
0.9547
],
"width_cells": [
3.6068,
35.1057
],
"depth_cells": [
3.0,
15.9999
],
"mCHsinu": [
1.1,
1.4
],
"mFFCHprop": [
0.1,
0.7
],
"probAvulInside": [
0.2,
0.55
]
},
"channel:SH_DISTAL": {
"ntg": [
0.0004,
0.9996
],
"width_cells": [
4.2059,
35.0671
],
"depth_cells": [
3.0,
15.9999
],
"mCHsinu": [
1.1,
1.45
],
"mFFCHprop": [
0.0,
0.2
],
"probAvulInside": [
0.02,
0.1
]
},
"channel:SH_PROXIMAL": {
"ntg": [
0.0,
0.9993
],
"width_cells": [
4.8072,
41.4372
],
"depth_cells": [
3.0,
15.9999
],
"mCHsinu": [
1.05,
1.4
],
"mFFCHprop": [
0.05,
0.4
],
"probAvulInside": [
0.2,
0.5
]
},
"channel:MEANDER_OXBOW": {
"ntg": [
0.0001,
0.9756
],
"width_cells": [
2.4096,
22.3713
],
"depth_cells": [
3.0,
15.9997
],
"mCHsinu": [
1.4,
1.85
],
"mFFCHprop": [
0.0,
0.0
],
"probAvulInside": [
0.0,
0.0
]
},
"delta": {
"ntg": [
0.0025,
0.8131
],
"width_cells": [
3.0148,
31.9141
],
"depth_cells": [
3.0,
15.9999
],
"mCHsinu": [
1.05,
1.1
],
"mFFCHprop": [
0.0,
0.4
],
"probAvulInside": [
0.5,
0.8
],
"trunk_length_fraction": [
0.1,
0.5
]
}
},
"sampler": {
"solver": "euler",
"steps": 100,
"guidance": 3.0,
"field_radius": 12
},
"checkpoint": "inference_epoch040.pt"
}