ucalyptus commited on
Commit
454970e
1 Parent(s): 95f8de4

Update tune.py

Browse files
Files changed (1) hide show
  1. tune.py +1 -1
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}/{image_name}'
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)