Skip to content

Instantly share code, notes, and snippets.

@marduk191
marduk191 / pytorch_to_safetensors.py
Created August 22, 2025 03:23
Robust_Pytorch_to_safetensors
#!/usr/bin/env python3
"""
Convert PyTorch files (.bin, .pth, .pt, .ckpt) to SafeTensors format.
By:marduk191
"""
import os
import argparse
import torch
from safetensors.torch import save_file, save_model
@marduk191
marduk191 / maxUpscaler_node.py
Created July 27, 2025 21:50
comfyui upscaler
import torch
import torch.nn.functional as F
import numpy as np
from PIL import Image, ImageFilter, ImageEnhance
import math
import comfy.model_management as mm
import comfy.utils
import comfy.samplers
import folder_paths
@marduk191
marduk191 / safetensor_combine.py
Created July 23, 2025 19:13
combine multiple clips and vae with the model for packaging
#!/usr/bin/env python3
"""
Script to merge separated CLIP, VAE, and model safetensor files back into a single safetensor file.
https://github.com/marduk191
"""
import os
import json
import struct
import torch
@marduk191
marduk191 / merge_shards.py
Created July 23, 2025 14:31
for merging shards like "model-00001-of-00005.safetensors"
"""
Safetensors deshard
https://github.com/marduk191
"""
import sys
import os
import json
import struct
import torch
from safetensors import safe_open
@marduk191
marduk191 / script_runner_gui.py
Created July 21, 2025 20:55
A small tkinter gui for loading python scripts from a directory.
import tkinter as tk
from tkinter import ttk, filedialog, messagebox, scrolledtext
import os
import subprocess
import threading
from pathlib import Path
import sys
class ScriptRunnerGUI:
def __init__(self, root):
#!/usr/bin/env python3
"""
Script to scan directory for safetensor files and extract model, clip, and vae components
into separate folders using memory-efficient loading.
https://github.com/marduk191
"""
import os
import json
import struct
#!/usr/bin/env python3
"""
SafeTensor Model Merger with Multiple Averaging Methods
Supports Simple Moving Average (SMA), Exponential Moving Average (EMA), and Weighted Moving Average (WMA)
https://github.com/marduk191
"""
import os
import sys
import json
@marduk191
marduk191 / pt_to_safe.py
Last active July 9, 2025 00:44 — forked from silveroxides/pt_to_safe.py
Pytorch Model to Safetensors converter
# created by https://gist.github.com/silveroxides
import torch
from safetensors.torch import save_file
import argparse
import os
def convert_pt_to_safetensors(model_path, safe_path):
model = torch.load(model_path, map_location="cpu", weights_only=False)
metadata = {"format":"pt"}
save_file(model, safe_path, metadata)
"""
https://github.com/marduk191
"""
import torch
import gc
from typing import Dict, Any, Tuple, Optional
import comfy.model_management as model_management
import comfy.utils
@marduk191
marduk191 / minimal_safetensors_converter.py
Last active July 21, 2025 00:51
Pt to safetensors converter with minimal checking. fp32 output
#!/usr/bin/env python3
"""
Minimal SafeTensors Converter
Converts PyTorch model files (.pt, .pth) to .safetensors format
Usage: python converter.py <input_path> [output_path]
by:marduk191
https://github.com/marduk191
"""
import os