Flatten model dirs: drop redundant seed0/ level (seed recorded in config.json)

README.md

@@ -58,7 +58,7 @@ model = build_model(
                 basis=BasisConfig(patch_size=32, p_max=16, s_e_range=(0.25, 2.0)),
                 encoder=EncoderConfig()),
     bandwidth_mode="global_scalar", adapt_order=False).eval()
-model.load_state_dict(load_file("gblsr-scalar/seed0/model.safetensors"))
+model.load_state_dict(load_file("gblsr-scalar/model.safetensors"))
 
 out = model(torch.rand(1, 3, 256, 256))   # dict of outputs
 recon = out["recon"]                       # (1, 3, 256, 256) reconstruction
@@ -74,13 +74,13 @@ from gblsr import GBLSRScalarASR
 # match encoder_cfg / decoder_cfg to the folder's config.json
 model = GBLSRScalarASR(encoder_cfg={"num_features": 96},
                        decoder_cfg={"local_ensemble": False}).eval()   # nf96+noLE
-model.load_state_dict(load_file("gblsr-scalar-asr-nf96-noLE/seed0/model.safetensors"))
+model.load_state_dict(load_file("gblsr-scalar-asr-nf96-noLE/model.safetensors"))
 
 lr = torch.rand(1, 3, 64, 64)
 hr = model.predict_full(lr, H_q=256, W_q=256)   # (1, 3, 256, 256), any target size
 ```
 
-Each model's exact build config is in its `seed0/config.json`.
+Each model's exact build config is in its `config.json`.
 
 ## Training
 

examples/infer.py

@@ -24,8 +24,8 @@ HERE = Path(__file__).resolve().parent.parent  # release root
 
 
 def load_model(name: str):
-    cfg = json.loads((HERE / name / "seed0" / "config.json").read_text())
-    weights = load_file(str(HERE / name / "seed0" / "model.safetensors"))
+    cfg = json.loads((HERE / name / "config.json").read_text())
+    weights = load_file(str(HERE / name / "model.safetensors"))
     if name == "gblsr-scalar":
         from gblsr import BasisConfig, EncoderConfig, ModelConfig, build_model