xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 12:54:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update curve_nodes.py

Browse Source
This commit is contained in:
Kijai 2024-04-29 18:33:10 +03:00
parent 6669c93efc
commit 9112777eb1
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nodes/curve_nodes.py
View File

@ -517,9 +517,9 @@ Experimental, does not function yet as ComfyUI base changes are needed
c = []
cond, cond_pooled = clip.encode_from_tokens(clip.tokenize(text), return_pooled=True)
image_height = latents['samples'].shape[-1] * 8
image_width = latents['samples'].shape[-2] * 8
plot_image_tensor = self.plot_coordinates_to_tensor(coordinates, image_height, image_width, height)
image_height = latents['samples'].shape[-2] * 8
image_width = latents['samples'].shape[-1] * 8
plot_image_tensor = self.plot_coordinates_to_tensor(coordinates, image_height, image_width, height, text)
for t in conditioning_to:
n = [t[0], t[1].copy()]
@ -544,7 +544,7 @@ Experimental, does not function yet as ComfyUI base changes are needed
return (c, plot_image_tensor,)
def plot_coordinates_to_tensor(self, coordinates, height, width, box_size):
def plot_coordinates_to_tensor(self, coordinates, height, width, box_size, prompt):
import matplotlib
matplotlib.use('Agg')
from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
@ -579,7 +579,7 @@ Experimental, does not function yet as ComfyUI base changes are needed
ax.set_ylabel('y', color='#999999')
for text in ax.get_xticklabels() + ax.get_yticklabels():
text.set_color('#999999')
ax.set_title('Gligen positions')
ax.set_title('Gligen pos for: ' + prompt)
ax.set_xlabel('X Coordinate')
ax.set_ylabel('Y Coordinate')
ax.legend().remove()