From 476602aa2f3bbb792ded560aa80aeace333011fc Mon Sep 17 00:00:00 2001
From: sakura1bgx <isakura710@gmail.com>
Date: Mon, 3 Mar 2025 04:28:10 +0900
Subject: [PATCH] Update flipstreamviewer.py

- Rename FlipStreamChatLoad to FlipStreamLoadChatModel
---
 flipstreamviewer.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/flipstreamviewer.py b/flipstreamviewer.py
index b2a66f9..3ccffa3 100644
--- a/flipstreamviewer.py
+++ b/flipstreamviewer.py
@@ -24,8 +24,7 @@ import comfy
 from nodes import CheckpointLoaderSimple, VAELoader
 
 try:
-    from llama_cpp import Llama
-    
+    from llama_cpp import Llama  # llama-cpp-python
 except:
     Llama = None
 
@@ -2334,7 +2333,7 @@ class FlipStreamSegMask:
         return (mask_tensor, mask_tensor[:,:,:,0])
 
 
-class FlipStreamChatLoad:
+class FlipStreamLoadChatModel:
     @classmethod
     def INPUT_TYPES(cls):
         return {
@@ -2600,7 +2599,7 @@ NODE_CLASS_MAPPINGS = {
     "FlipStreamGate": FlipStreamGate,
     "FlipStreamRembg": FlipStreamRembg,
     "FlipStreamSegMask": FlipStreamSegMask,
-    "FlipStreamChatLoad": FlipStreamChatLoad,
+    "FlipStreamLoadChatModel": FlipStreamLoadChatModel,
     "FlipStreamChat": FlipStreamChat,
     "FlipStreamBatchPrompt": FlipStreamBatchPrompt,
     "FlipStreamFilmVfi": FlipStreamFilmVfi,
@@ -2637,7 +2636,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "FlipStreamGate": "FlipStreamGate",
     "FlipStreamRembg": "FlipStreamRembg",
     "FlipStreamSegMask": "FlipStreamSegMask",
-    "FlipStreamChatLoad": "FlipStreamChatLoad",
+    "FlipStreamLoadChatModel": "FlipStreamLoadChatModel",
     "FlipStreamChat": "FlipStreamChat",
     "FlipStreamBatchPrompt": "FlipStreamBatchPrompt",
     "FlipStreamFilmVfi": "FlipStreamFilmVfi",