Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

belka_gate_fusions: TODO: fix Incorrect fusion behaviour

Browse Source
This commit is contained in:
pikaliov
2026-05-19 13:30:25 +03:00
parent a2729c065e
commit ac9cdaf6f5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/models/asymmetric_encoder.py
View File

@@ -297,6 +297,7 @@ class AsymmetricEncoder(nn.Module):
def __init__( def __init__(
self, self,
# !!! ---------------------------------------------------------------
gate_type: ResidualGateType = ResidualGateType.simple_residual_one_gate, gate_type: ResidualGateType = ResidualGateType.simple_residual_one_gate,
dino_web_path: str = "nn_models/DINO_WEB/dinov3-vitl16-pretrain-lvd1689m.pth", dino_web_path: str = "nn_models/DINO_WEB/dinov3-vitl16-pretrain-lvd1689m.pth",
dino_sat_path: str = "nn_models/DINO_SAT/model.safetensors", dino_sat_path: str = "nn_models/DINO_SAT/model.safetensors",