mimbres commited on
Commit
3e98930
1 Parent(s): acbafd7

Update app.py

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