fix path in test

linoytsaban · linoytsaban · commit 31de75262016 · 2024-10-16T11:47:58.000+03:00
diff --git a/examples/advanced_diffusion_training/test_dreambooth_lora_flux_advanced.py b/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"{tmpdir}_emb.safetensors")
+                os.path.join(tmpdir, f"{os.path(tmpdir).name}_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)

Original file line number	Diff line number	Diff line change
`@@ -178,7 +178,7 @@ def test_dreambooth_lora_pivotal_tuning_flux_clip_t5(self):`
`178`	`178`
`179`	`179`	`# make sure the state_dict has the correct naming in the parameters.`
`180`	`180`	`textual_inversion_state_dict = safetensors.torch.load_file(`
`181`		`- os.path.join(tmpdir, f"{tmpdir}_emb.safetensors")`
	`181`	`+ os.path.join(tmpdir, f"{os.path(tmpdir).name}_emb.safetensors")`
`182`	`182`	`)`
`183`	`183`	`is_te = all(("clip_l" in k or "t5" in k) for k in textual_inversion_state_dict.keys())`
`184`	`184`	`self.assertTrue(is_te)`