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