Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ elif model_name == "YPTF+Single (noPS)":
|
|
32 |
args = [checkpoint, '-p', project, '-enc', 'perceiver-tf', '-ac', 'spec',
|
33 |
'-hop', '300', '-atc', '1', '-pr', precision]
|
34 |
elif model_name == "YPTF+Multi (PS)":
|
35 |
-
checkpoint = "
|
36 |
args = [checkpoint, '-p', project, '-tk', 'mc13_full_plus_256',
|
37 |
'-dec', 'multi-t5', '-nl', '26', '-enc', 'perceiver-tf',
|
38 |
'-ac', 'spec', '-hop', '300', '-atc', '1', '-pr', precision]
|
|
|
32 |
args = [checkpoint, '-p', project, '-enc', 'perceiver-tf', '-ac', 'spec',
|
33 |
'-hop', '300', '-atc', '1', '-pr', precision]
|
34 |
elif model_name == "YPTF+Multi (PS)":
|
35 |
+
checkpoint = "mc13_256_all_cross_v6_xk5_amp0811_edr005_attend_c_full_plus_2psn_nl26_sb_b26r_800k@model.ckpt"
|
36 |
args = [checkpoint, '-p', project, '-tk', 'mc13_full_plus_256',
|
37 |
'-dec', 'multi-t5', '-nl', '26', '-enc', 'perceiver-tf',
|
38 |
'-ac', 'spec', '-hop', '300', '-atc', '1', '-pr', precision]
|