Upload MPTransformer model

README.md

@@ -4,12 +4,12 @@ tags:
 - pytorch
 - robotics
 - custom-code
-library_name: nmp
+library_name: drp
 ---
 
 # MPTransformer
 
-This repository contains an `MPTransformer` model exported from the `nmp` codebase.
+This repository contains an `MPTransformer` model exported from the `drp` codebase.
 
 ## Files
 
@@ -20,7 +20,7 @@ This repository contains an `MPTransformer` model exported from the `nmp` codeba
 ## Load
 
 ```python
-from nmp.model.transformer import MPTransformer
+from drp.model.transformer import MPTransformer
 
 model = MPTransformer.from_pretrained("jimyoung6709/DRP")
 model.eval()
@@ -32,4 +32,4 @@ Exported from `best.pt`.
 
 ## Notes
 
-This model depends on the local `nmp` package and project-specific dependencies such as `pointnet2_ops`.
+This model depends on the local `drp` package and project-specific dependencies such as `pointnet2_ops`.