Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ log_file = 'amt/log.txt'
|
|
23 |
Path(log_file).touch()
|
24 |
|
25 |
# @title Load Checkpoint
|
26 |
-
model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
|
27 |
precision = '16'# if torch.cuda.is_available() else '32'# @param ["32", "bf16-mixed", "16"]
|
28 |
project = '2024'
|
29 |
|
|
|
23 |
Path(log_file).touch()
|
24 |
|
25 |
# @title Load Checkpoint
|
26 |
+
model_name = "YMT3+" # 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
|
27 |
precision = '16'# if torch.cuda.is_available() else '32'# @param ["32", "bf16-mixed", "16"]
|
28 |
project = '2024'
|
29 |
|