Update tune.py
Browse files
    	
        tune.py
    CHANGED
    
    | @@ -38,7 +38,7 @@ def tensor_to_pil(img): | |
| 38 | 
             
            def tune():
         | 
| 39 | 
             
                model_id = run_PTI(run_name='',use_wandb=False, use_multi_id_training=False)
         | 
| 40 | 
             
                w_path_dir = f'{paths_config.embedding_base_dir}/{paths_config.input_data_id}'
         | 
| 41 | 
            -
                embedding_dir = f'{w_path_dir}/{paths_config.pti_results_keyword}/ | 
| 42 | 
             
                w_pivot = torch.load(f'{embedding_dir}/0.pt')
         | 
| 43 | 
             
                new_G = load_generator(model_id)
         | 
| 44 | 
             
                new_image = new_G.synthesis(w_pivot, noise_mode='const', force_fp32 = True)
         | 
|  | |
| 38 | 
             
            def tune():
         | 
| 39 | 
             
                model_id = run_PTI(run_name='',use_wandb=False, use_multi_id_training=False)
         | 
| 40 | 
             
                w_path_dir = f'{paths_config.embedding_base_dir}/{paths_config.input_data_id}'
         | 
| 41 | 
            +
                embedding_dir = f'{w_path_dir}/{paths_config.pti_results_keyword}/file'
         | 
| 42 | 
             
                w_pivot = torch.load(f'{embedding_dir}/0.pt')
         | 
| 43 | 
             
                new_G = load_generator(model_id)
         | 
| 44 | 
             
                new_image = new_G.synthesis(w_pivot, noise_mode='const', force_fp32 = True)
         | 
