fix filename of embedding

Author: linoytsaban

examples/advanced_diffusion_training/test_dreambooth_lora_flux_advanced.py

@@ -178,7 +178,7 @@ def test_dreambooth_lora_pivotal_tuning_flux_clip_t5(self):
 
             # make sure the state_dict has the correct naming in the parameters.
             textual_inversion_state_dict = safetensors.torch.load_file(
-                os.path.join(tmpdir, f"{os.path(tmpdir).name}_emb.safetensors")
+                os.path.join(tmpdir, f"{os.path.basename(tmpdir)}_emb.safetensors")
             )
             is_te = all(("clip_l" in k or "t5" in k) for k in textual_inversion_state_dict.keys())
             self.assertTrue(is_te)

examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py

@@ -2405,7 +2405,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
             )
 
         if args.train_text_encoder_ti:
-            embeddings_path = f"{args.output_dir}/{args.output_dir}_emb.safetensors"
+            embeddings_path = f"{args.output_dir}/{os.path.basename(args.output_dir)}_emb.safetensors"
             embedding_handler.save_embeddings(embeddings_path)
 
         # Final inference