Update Era_s20_updt.py
Browse files- 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 |
|