ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 05:08:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_comfyui.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-16 01:37:21 +02:00 committed by GitHub
parent 156968df95
commit ef277036dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lollms/services/comfyui/lollms_comfyui.py
View File

@ -393,9 +393,9 @@ class LollmsComfyUI(LollmsTTI):
prompt["1"]["inputs"]["base_ckpt_name"] = self.app.config.comfyui_model prompt["1"]["inputs"]["base_ckpt_name"] = self.app.config.comfyui_model
ws = websocket.WebSocket() ws = websocket.WebSocket()
ws.connect("ws://{}/ws?clientId={}".format(self.comfyui_base_url, client_id)) ws.connect("ws://{}/ws?clientId={}".format(self.comfyui_base_url[7:-1], client_id))
images = get_images(ws, prompt) images = get_images(ws, prompt)
return None return images
def paint_from_images(self, positive_prompt: str, def paint_from_images(self, positive_prompt: str,
images: List[str], images: List[str],