jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Add route to get safetensors metadata:

Browse Source 
/view_metadata/loras?filename=lora.safetensors
This commit is contained in:
comfyanonymous
2023-05-29 02:48:50 -04:00
parent 23ffafeb5d
commit b9818eb910
3 changed files with 35 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
server.py
View File

@@ -22,7 +22,7 @@ except ImportError:
import mimetypes
from comfy.cli_args import args
import comfy.utils
@web.middleware
async def cache_control(request: web.Request, handler):
@@ -257,6 +257,29 @@ class PromptServer():
return web.Response(status=404)
@routes.get("/view_metadata/{folder_name}")
async def view_metadata(request):
folder_name = request.match_info.get("folder_name", None)
if folder_name is None:
return web.Response(status=404)
if not "filename" in request.rel_url.query:
return web.Response(status=404)
filename = request.rel_url.query["filename"]
if not filename.endswith(".safetensors"):
return web.Response(status=404)
safetensors_path = folder_paths.get_full_path(folder_name, filename)
if safetensors_path is None:
return web.Response(status=404)
out = comfy.utils.safetensors_header(safetensors_path, max_size=1024*1024)
if out is None:
return web.Response(status=404)
dt = json.loads(out)
if not "__metadata__" in dt:
return web.Response(status=404)
return web.json_response(dt["__metadata__"])
@routes.get("/prompt")
async def get_prompt(request):
return web.json_response(self.get_queue_info())