Create pytorch_model.bin
Browse files- pytorch_model.bin +7 -0
pytorch_model.bin
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
|
| 3 |
+
# Load the Aiden_t5 model weights
|
| 4 |
+
aidenT5Model = torch.load('/path/to/model_weights/pytorch_model.bin')
|
| 5 |
+
|
| 6 |
+
# Initialize the Aiden_t5 model
|
| 7 |
+
aidenT5Model = Aiden_t5(aidenT5Model)
|