nkanungo commited on
Commit
30fbdbc
·
1 Parent(s): 6478217

Update Era_s20_updt.py

Browse files
Files changed (1) hide show
  1. Era_s20_updt.py +1 -1
Era_s20_updt.py CHANGED
@@ -87,7 +87,7 @@ vae = vae.to(torch_device)
87
  text_encoder = text_encoder.to(torch_device)
88
  unet = unet.to(torch_device);
89
 
90
- embeds_folder = Path(paintings_embed)
91
  file_names = [path.name for path in embeds_folder.glob('*') if path.is_file()]
92
  print(file_names)
93
 
 
87
  text_encoder = text_encoder.to(torch_device)
88
  unet = unet.to(torch_device);
89
 
90
+ embeds_folder = Path('./paintings_embed')
91
  file_names = [path.name for path in embeds_folder.glob('*') if path.is_file()]
92
  print(file_names)
93