Add model card
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ from train import load_model_from_huggingface
|
|
| 40 |
import torch
|
| 41 |
|
| 42 |
# Load the model
|
| 43 |
-
model = load_model_from_huggingface("CatkinChen/nethack-vae-hmm")
|
| 44 |
|
| 45 |
# Example usage with synthetic data
|
| 46 |
batch_size = 1
|
|
|
|
| 40 |
import torch
|
| 41 |
|
| 42 |
# Load the model
|
| 43 |
+
model, config = load_model_from_huggingface("CatkinChen/nethack-vae-hmm")
|
| 44 |
|
| 45 |
# Example usage with synthetic data
|
| 46 |
batch_size = 1
|