toolkit / inspecting_state_dict.py
k4d3's picture
Fix trailing newline in inspecting_state_dict.py for consistency
b4b77ff
raw
history blame contribute delete
180 Bytes
import safetensors.torch
state_dict = safetensors.torch.load_file("/home/kade/source/repos/JTP2/JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
print(state_dict.keys())