up

convert_flax_to_pt.py

@@ -2,6 +2,7 @@ import argparse
 import json
 import os
 import shutil
+from diffusers.pipelines.stable_diffusion import safety_checker
 import torch
 from tempfile import TemporaryDirectory
 from typing import List, Optional
@@ -18,7 +19,7 @@ def convert(api: "HfApi", model_id: str, force: bool = False) -> Optional["Commi
     is_sd = "model_index.json" in filenames
 
     if is_sd:
-        model = StableDiffusionPipeline.from_pretrained(model_id, from_flax=True)
+        model = StableDiffusionPipeline.from_pretrained(model_id, from_flax=True, safety_checker=None)
     else:
         model = ControlNetModel.from_pretrained(model_id, from_flax=True)
 

mass_open_controlnet_pr.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+while read p; do
+	echo "-------------------------------"
+	echo "Open PR for $p"
+	python convert_flax_to_pt.py $p
+done