- load_checkpoint.py +2 -0
load_checkpoint.py
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
# @title Load Checkpoint
|
|
|
|
|
|
|
| 2 |
model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
|
| 3 |
precision = '16' # @param ["32", "bf16-mixed", "16"]
|
| 4 |
project = '2024'
|
|
|
|
| 1 |
# @title Load Checkpoint
|
| 2 |
+
from model_helper import laod_model_checkpoint
|
| 3 |
+
|
| 4 |
model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
|
| 5 |
precision = '16' # @param ["32", "bf16-mixed", "16"]
|
| 6 |
project = '2024'
|