temp

sgm/modules/diffusionmodules/openaimodel.py

@@ -2021,7 +2021,7 @@ class UNetAddModel(nn.Module):
             sns.heatmap(attn_map_i[j], square=True, xticklabels=False, yticklabels=False)
             if j < len(tokens):
                 ax.set_title(tokens[j])
-        fig.savefig(f"temp/attn_map/attn_map_{save_name}.png")
+        fig.savefig(f"./temp/attn_map/attn_map_{save_name}.png")
         plt.close()
 
         return attn_map_i

sgm/modules/diffusionmodules/sampling.py

@@ -289,9 +289,9 @@ class EulerEDMSampler(EDMSampler):
             sections.append(attn_map)
         
         section = np.stack(sections)
-        np.save(f"temp/seg_map/seg_{save_name}.npy", section)
+        np.save(f"./temp/seg_map/seg_{save_name}.npy", section)
 
-        save_image(image_, f"temp/seg_map/seg_{save_name}.png", normalize=True)
+        save_image(image_, f"./temp/seg_map/seg_{save_name}.png", normalize=True)
 
     def get_init_noise(self, cfgs, model, cond, batch, uc=None):