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