from safetensors.torch import load_file

# Load the safetensors file
filename = "flowgram.safetensors"
tensor_dict = load_file(filename)

# Open a log file to write the keys and dtype
with open('log.txt', 'w') as log_file:
    for key, tensor in tensor_dict.items():
        # Write the key and the dtype of the tensor
        log_file.write(f"Key: {key}, Dtype: {tensor.dtype}\n")

print("Log file has been updated with keys and dtype.")