Update model.py
Browse files
model.py
CHANGED
|
@@ -9,7 +9,7 @@ from jaxtyping import Float
|
|
| 9 |
from huggingface_hub import PyTorchModelHubMixin
|
| 10 |
|
| 11 |
|
| 12 |
-
class Model(nn.Module, PyTorchModelHubMixin, repo_url="https://huggingface.co/TimWalter/RAM/new/main",
|
| 13 |
"""
|
| 14 |
RAM model that predicts reachability from a modified Denavit-Hartenberg morphology parametrisation and a pose.
|
| 15 |
"""
|
|
|
|
| 9 |
from huggingface_hub import PyTorchModelHubMixin
|
| 10 |
|
| 11 |
|
| 12 |
+
class Model(nn.Module, PyTorchModelHubMixin, repo_url="https://huggingface.co/TimWalter/RAM/new/main", paper_url="https://huggingface.co/papers/2606.09108",license="mit",):
|
| 13 |
"""
|
| 14 |
RAM model that predicts reachability from a modified Denavit-Hartenberg morphology parametrisation and a pose.
|
| 15 |
"""
|