Seeking help in the creation of a ComfyUI CUSTOM NODE.

[moreAImore]

Hello dear PULID creator!
I am trying to modify an existing custom node for comfyui to make it able to run fp8 just like your local gradio:
I made these changes (see 3 latest commits)
here: https://github.com/moreAImore/ComfyUI-PuLID-Flux-Enhanced-FP8

I have ONE PROBLEM:
When I run it first time it IS fast. BUT when I run it a second time, it uses MORE GPU, and my pc reach PEAK gpu usage (it becomes slower way slower)
Anywho the problem is extra gpu usage.

Could you see if you can find what is missing and is causing the code to use more gpu the second time the image generation is run whereas first generation is very fast?

Thanks