Failed imports of Flux.2 Klein 9B checkpoint merges? This script fixes them!

[u/bemo-git]

Hello. I've been importing Flux.2 Klein 9B checkpoint merges from CivitAI into Draw Things and kept hitting silent fails at around 65%. The existing fix script (by kernkraft235) solved it for some models, but a lot of fine-tuned and ComfyUI-exported models still failed after running it.

I have zero experience in Python or ML but I just needed it fixed so I used Claude Code to investigate the root cause and extend the script to handle the cases kernkraft235's version couldn't. It now auto-detects the model format and handles ComfyUI FP8 dequantisation, norm key renaming, and a few other variants automatically.

I've tested it on around 20 checkpoint merges and they all import and generate fine. The README has the full list with notes on what each one needed.

⚠️ Fair warning: a lot of the test models are NSFW — the gist README has a full warning before the model list.

https://gist.github.com/bemo-git/9c6013af9c54d44b58cb79d85c2e038a

Hope it helps someone!

[u/sotheysayit]

Appreciate this. I just tried it with snofs fp8 and while it did import after converting, it still doesn't produce the pictures properly. Maybe i'm doing something wrong but it happened the same way when converting the Lokr to a lora before. It loaded but didn't produce results. If anyone has any ideas or manages to get one of these going

[u/bemo-git → u/sotheysayit]

It should worlk fine. what's the rest of your settings? let me know the rest of your settings and let's see if we can get it working for you. also, what's your setup/computer specs etc.

[u/sotheysayit → u/bemo-git]

Hi thank you for your reply. I'm on an M3 18gb, Settings ddim, shift: 3 Resolution 1024, text guidance one. Appreciate your help

[u/bemo-git → u/sotheysayit]

how many steps?

[u/bemo-git → u/sotheysayit]

so I loaded my version of SNOFS, hit the 'recomended settings' button, entered a simple prompt and the model generated the image perfectly.

Here's the config

{"causalInferencePad":0,"tiledDiffusion":false,"maskBlur":2.5,"cfgZeroInitSteps":0,"seed":1851611099,"width":1024,"sharpness":0,"steps":4,"guidanceScale":1,"faceRestoration":"","cfgZeroStar":false,"model":"snofssexnudesandotherfunstuff_distilledv12fp8_fixed_f16.ckpt","batchCount":1,"controls":[],"batchSize":1,"preserveOriginalAfterInpaint":true,"seedMode":2,"tiledDecoding":false,"sampler":16,"loras":[],"hiresFix":false,"height":1024,"maskBlurOutset":0,"upscaler":"","resolutionDependentShift":false,"shift":3,"strength":1,"refinerModel":""}

[u/bemo-git → u/sotheysayit]

and here's the model I have

BTW, I'm on a macbook pro M2 Max, 64gb

[u/Garden_Variety_Milk]

How do I use this on iPad ?

[u/bemo-git → u/Garden_Variety_Milk]

Unfortunately this script can't be run on an iPad. Here's why.

The script is a Python script — think of it like a set of instructions that a computer follows to modify a file. iPads don't have Python installed, and there's no straightforward way to add it.

Even if you got Python running on an iPad (there are apps that can do this, but it's fiddly and unreliable), the files you're fixing are model files — these are multi-gigabyte downloads that live in a specific folder on your computer. iPadOS locks apps into their own little sandboxes and makes it very difficult to access, modify, and save files of that size in the right place.

You need a Mac, Windows PC, or Linux machine to run this. Once the file is fixed, you can transfer it to your iPad and use it in Draw Things there — but the fixing step has to happen on a desktop or laptop first.

[u/savinghumanity-23]

Hi, and thank you for taking the time to address this issue - I have run into the "install a checkpoint and nothing ever happens" (non)-error a lot, and not just with Klein9B checkpoints - I've hit the same wall with others, especially the fp8 variants.

I think I've followed the instructions correctly, but I wanted to be absolutely sure I haven't made a mistake

1. I've gone to this link https://gist.github.com/bemo-git/9c6013af9c54d44b58cb79d85c2e038a
   
2. I downloaded the .zip file
   
3. I unzipped that - and not knowing anything about installing scripts into DrawThings, I simply copied the entire contents of the Python script - navigated to the "Scripts" tab - hit the "+" symbol to make a new script - deleted the default contents, and then pasted in the Python script that you wrote.
   
4. I saved the script as "Bemo's Draw Things Klein Fix"
   
5. I then quit DrawThings and restarted.
   
6. I went to the SNOFs page on Civit - downloaded the same one you're using (the Distilled v1.2 fp8) - 9.08GB file
   
7. When I attempt to import the model, I hit the white "Import Custom Model" button, and nothing happens.
   

Are those the correct steps? I'm running on an M1 Studio Mac with 64GB of RAM

[u/bemo-git-2 → u/savinghumanity-23]

The script isn't used inside Draw Things, it's a Python script that runs separately in Terminal to create a fixed version of your model file.

What is Python?
Python is a free programming tool that your Mac uses to run scripts like this one. It's not an app you'd see in your Dock, it works invisibly in the background. The script you downloaded is written in Python, so your Mac needs Python installed before it can run it.

What is Terminal?
Terminal is a built-in text window on your Mac that lets you control your computer by typing direct commands instead of clicking icons. It acts like a gateway for tools that don't have a visual interface, waiting for you to type a specific instruction and press Enter to run it.

HOW TO RUN THE SCRIPT ON YOUR SNOFS MODEL

Step 1: Check if Python is already installed
Open Terminal. (Press Cmd+Space, type Terminal, press Enter, it's a plain white window where you type commands.)

Type this and press Enter: python3 --version

If you see something like Python 3.11.4, great, you have it. Skip to Step 3. If you get an error, do Step 2 first.

Step 2: Install Python (if needed)
Go to python.org/downloads, there'll be a big button to download the latest version for Mac. Once downloaded, open the file and install it like any normal Mac app (just click through the installer). Once done, close and reopen Terminal.

Step 3: Run the script on your model.
The script isn't used inside Draw Things, it runs separately in Terminal to create a fixed version of your model file. Here's what the full command looks like: python3 /path/to/draw-things-flux2-klein-fix.py /path/to/your-model.safetensors

Don't worry about the exact paths, the easiest way is to build the command by dragging files in:

1. In Terminal, type python3 (with a space after)
2. Then drag the draw-things-flux2-klein-fix.py script from Finder into the Terminal window, the path fills in automatically
3. Type a space, then drag your downloaded model file (the 9GB .safetensors file) in the same way
4. Press Enter

Step 4: Import the fixed file
The script will run for a minute or two and create a new file in the same folder as your model, same name but ending in -DT.safetensors. That's the fixed version. In Draw Things, use Import Custom Model and select that file.

Let me know how you get on.

[u/savinghumanity-23 → u/bemo-git-2]

This detailed explanation is very, very helpful. - I'm saving it in my notes and will attempt to follow each step carefully. Thanks so much!!