1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-03 07:26:31 +08:00

Add route to get safetensors metadata:

/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

View File

@@ -1,5 +1,6 @@
import torch
import math
import struct
def load_torch_file(ckpt, safe_load=False):
if ckpt.lower().endswith(".safetensors"):
@@ -50,6 +51,14 @@ def transformers_convert(sd, prefix_from, prefix_to, number):
sd[k_to] = weights[shape_from*x:shape_from*(x + 1)]
return sd
def safetensors_header(safetensors_path, max_size=100*1024*1024):
with open(safetensors_path, "rb") as f:
header = f.read(8)
length_of_header = struct.unpack('<Q', header)[0]
if length_of_header > max_size:
return None
return f.read(length_of_header)
def bislerp(samples, width, height):
def slerp(b1, b2, r):
'''slerps batches b1, b2 according to ratio r, batches should be flat e.g. NxC'''