JTriggerFish

import re
import pandas as pd
import torch
from safetensors import torch as storch


def parse_layer_name(name: str):
    """Same as before: parse (component, sub_block, param_type). Adjust as needed."""
    if "model.diffusion_model" in name:
        component = "unet"

JTriggerFish

{
 "metadata": {
  "name": "Tanh approximations"
 },
 "nbformat": 3,
 "nbformat_minor": 0,
 "worksheets": [
  {
   "cells": [
    {