diff --git a/import_schemas.py b/import_schemas.py old mode 100644 new mode 100755 index 5027d08..97a4a92 --- a/import_schemas.py +++ b/import_schemas.py @@ -1,45 +1,8 @@ +#!/usr/bin/env python3 import replicate import json import os -models_to_import = [ - "andreasjansson/blip-2", - "bytedance/sdxl-lightning-4step", - "fofr/face-to-many", - "fofr/sd3-with-chaos", - "meta/llama-2-7b-chat", - "meta/llama-2-70b-chat", - # "meta/meta-llama-3-70b-instruct", - # "meta/meta-llama-3-8b-instruct", - "mistralai/mixtral-8x7b-instruct-v0.1", - "philz1337x/clarity-upscaler", - "salesforce/blip", - "smoretalk/rembg-enhance", - "stability-ai/sdxl", - "stability-ai/stable-diffusion-3", - "yorickvp/llava-13b", - "yorickvp/llava-v1.6-34b", - "yorickvp/llava-v1.6-mistral-7b", - "snowflake/snowflake-arctic-instruct", - "batouresearch/high-resolution-controlnet-tile", - "batouresearch/magic-style-transfer", - "batouresearch/magic-image-refiner", - "lucataco/pasd-magnify", - "cjwbw/supir", - "lucataco/qwen-vl-chat", - "omniedgeio/face-swap", - "ai-forever/kandinsky-2.2", - "lucataco/sdxl-clip-interrogator", - "lucataco/paligemma-3b-pt-224", - "lucataco/llama-3-vision-alpha", - "cuuupid/idm-vton", - "cuuupid/glm-4v-9b", - "lucataco/moondream2", - "cjwbw/hyper-sdxl-1step-t2i", - "cjwbw/pixart-sigma", - "lucataco/hunyuandit-v1.1", -] - def format_json_file(file_path): try: @@ -62,7 +25,10 @@ def format_json_files_in_directory(directory): format_json_file(file_path) -for model in models_to_import: +with open("supported_models.json", "r") as f: + supported_models = json.load(f) + +for model in supported_models["models"]: m = replicate.models.get(model) with open(f"schemas/{model.replace('/', '_')}.json", "w") as f: f.write(m.json()) diff --git a/supported_models.json b/supported_models.json new file mode 100644 index 0000000..78a5ee4 --- /dev/null +++ b/supported_models.json @@ -0,0 +1,39 @@ +{ + "models": [ + "ai-forever/kandinsky-2.2", + "andreasjansson/blip-2", + "batouresearch/high-resolution-controlnet-tile", + "batouresearch/magic-image-refiner", + "batouresearch/magic-style-transfer", + "bytedance/sdxl-lightning-4step", + "cjwbw/hyper-sdxl-1step-t2i", + "cjwbw/pixart-sigma", + "cjwbw/supir", + "cuuupid/glm-4v-9b", + "cuuupid/idm-vton", + "fofr/consistent-character", + "fofr/controlnet-preprocessors", + "fofr/face-to-many", + "fofr/sd3-with-chaos", + "lucataco/hunyuandit-v1.1", + "lucataco/llama-3-vision-alpha", + "lucataco/moondream2", + "lucataco/paligemma-3b-pt-224", + "lucataco/pasd-magnify", + "lucataco/qwen-vl-chat", + "lucataco/sdxl-clip-interrogator", + "meta/llama-2-70b-chat", + "meta/llama-2-7b-chat", + "mistralai/mixtral-8x7b-instruct-v0.1", + "omniedgeio/face-swap", + "philz1337x/clarity-upscaler", + "salesforce/blip", + "smoretalk/rembg-enhance", + "snowflake/snowflake-arctic-instruct", + "stability-ai/sdxl", + "stability-ai/stable-diffusion-3", + "yorickvp/llava-13b", + "yorickvp/llava-v1.6-34b", + "yorickvp/llava-v1.6-mistral-7b" + ] +}