Upload folder using huggingface_hub

model_pole_position.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:62d218b9859acd4d19cfcfe6b3aa93ae129485a872175632ed32d6441ae9c7f6
-size 1580934
+oid sha256:8e0affcef8e533a29037751e27948a3eb0f2fda2792ce2b3dfc876cadb09e281
+size 2971526

model_pong_direct.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6ada51c7d09e003a2bea134bfe7be0e762756f10cebaab73c904135c5a4e33cf
-size 2437546
+oid sha256:381e3abbf7c5c308e985f53a916379140be588383760fa117009775b6bc79281
+size 1262522

predict.py

@@ -1,4 +1,4 @@
-"""Optimized blend: Pong AR weight 0.85->0.65, Sonic unchanged 0.7->0.3."""
+"""Full PP swap: Pong direct int8, full PP model, Sonic AR fp16 + direct int8."""
 import sys
 import os
 import numpy as np
@@ -65,13 +65,12 @@ def load_model(model_dir: str):
     pong.eval()
     ens.models["pong"] = pong
 
-    # Pong direct (fp16, 24 outputs)
+    # Pong direct (int8 quantized, 24 outputs)
     pong_direct = UNet(in_channels=24, out_channels=24,
                        enc_channels=(32, 64, 128), bottleneck_channels=128,
                        upsample_mode="bilinear").to(DEVICE)
-    sd = torch.load(os.path.join(model_dir, "model_pong_direct.pt"),
-                    map_location=DEVICE, weights_only=True)
-    pong_direct.load_state_dict({k: v.float() for k, v in sd.items()})
+    sd = load_int8_state_dict(os.path.join(model_dir, "model_pong_direct.pt"), DEVICE)
+    pong_direct.load_state_dict(sd)
     pong_direct.eval()
     ens.pong_direct = pong_direct
 
@@ -94,9 +93,9 @@ def load_model(model_dir: str):
     sonic_direct.eval()
     ens.sonic_direct = sonic_direct
 
-    # PP compact direct (fp16, 24 outputs)
+    # PP full direct (fp16, 24 outputs)
     pp = UNet(in_channels=24, out_channels=24,
-              enc_channels=(24, 48, 96), bottleneck_channels=128,
+              enc_channels=(32, 64, 128), bottleneck_channels=192,
               upsample_mode="bilinear").to(DEVICE)
     sd = torch.load(os.path.join(model_dir, "model_pole_position.pt"),
                     map_location=DEVICE, weights_only=True)