Richhiey commited on
Commit
7806391
1 Parent(s): 6d5bae9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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