Update README.md

README.md

@@ -8,6 +8,43 @@ tags:
   - geolip
 ---
 
+# Automodel Now Available
+```
+# ── Cell 1: Upload the fixed modeling file ──
+from huggingface_hub import upload_file
+
+upload_file(
+    path_or_fileobj="modeling_flow_match.py",
+    path_in_repo="modeling_flow_match.py",
+    repo_id="AbstractPhil/geolip-diffusion-proto",
+    commit_message="Fix HF PreTrainedModel attributes — _tied_weights_keys, post_init",
+)
+print("✓ Updated")
+
+
+# ── Cell 2: Test AutoModel loading + generation ──
+from transformers import AutoModel
+from torchvision.utils import save_image, make_grid
+
+model = AutoModel.from_pretrained(
+    "AbstractPhil/geolip-diffusion-proto", trust_remote_code=True
+).cuda()
+
+print(f"Params: {sum(p.numel() for p in model.parameters()):,}")
+print(f"Relay diagnostics: {model.get_relay_diagnostics()}")
+
+# Generate per-class samples
+class_names = ['plane','auto','bird','cat','deer','dog','frog','horse','ship','truck']
+all_imgs = []
+for c in range(10):
+    imgs = model.sample(n_samples=4, class_label=c)
+    all_imgs.append(imgs)
+
+grid = make_grid(torch.cat(all_imgs), nrow=10)
+save_image(grid, "automodel_test.png")
+print("✓ Saved automodel_test.png — 4 per class, 10 classes")
+```
+
 # GeoLIP Diffusion Prototype
 
 **Flow matching diffusion with constellation relay as geometric regulator.**