harinag0204 commited on
Commit
333238e
·
verified ·
1 Parent(s): 6e23e7e

Upload MPHGNet v4.1 model + weights

Browse files
Files changed (3) hide show
  1. README.md +36 -0
  2. model_weights.pt +3 -0
  3. mphgnet_v41.py +0 -0
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - physics
5
+ - materials-science
6
+ - graph-neural-network
7
+ - equivariant
8
+ license: mit
9
+ ---
10
+
11
+ # MPHGNet v4.1
12
+
13
+ Multi-Physics Hierarchical Graph Network (MPHGNet) v4.1 for grain-boundary energy,
14
+ forces, stress and multi-fidelity materials property prediction.
15
+
16
+ ## Model Description
17
+ - Equivariant physics backbone with e3nn irreps
18
+ - Multi-physics channels: local density, gradient flow, interaction energy,
19
+ global context, structural stability, Hilbert-space coherence, defect, diffusion, phonon
20
+ - Hebbian STDP plasticity module
21
+ - Triplet + quadruplet physics message passing
22
+
23
+ ## Usage
24
+ ```python
25
+ from huggingface_hub import hf_hub_download
26
+ import torch, importlib.util, sys
27
+
28
+ path = hf_hub_download(repo_id="harinag0204/MPHGNet-v41", filename="mphgnet_v41.py")
29
+ spec = importlib.util.spec_from_file_location("mphgnet_v41", path)
30
+ mph = importlib.util.module_from_spec(spec); spec.loader.exec_module(mph)
31
+
32
+ model = mph.EquivariantPhysicsBackboneV41(num_layers=1).to(mph.device)
33
+ weights = hf_hub_download(repo_id="harinag0204/MPHGNet-v41", filename="model_weights.pt")
34
+ model.load_state_dict(torch.load(weights, map_location=mph.device))
35
+ model.eval()
36
+ ```
model_weights.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edc2ce0ce7ba0e6994c31b50ea1ceaf2e70430a45d0ee7c7985a7c13f3e553d3
3
+ size 21275335
mphgnet_v41.py ADDED
The diff for this file is too large to render. See raw diff