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 |  | 
