mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-18 22:08:13 +00:00
21 lines
963 B
Python
21 lines
963 B
Python
import os
|
|
|
|
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
def get_model_path(model_name):
|
|
return os.path.join(CURRENT_DIR, MODEL_FOLDER, model_name)
|
|
|
|
MODEL_FOLDER = "reward_pretrained"
|
|
|
|
MODEL_PATHS = {
|
|
"aesthetic_predictor": get_model_path("aesthetic-predictor/sac+logos+ava1-l14-linearMSE.safetensors"),
|
|
"open_clip": get_model_path("CLIP-ViT-H-14-laion2B-s32B-b79K/open_clip_pytorch_model.bin"),
|
|
"hpsv2": get_model_path("HPS_v2/HPS_v2_compressed.safetensors"),
|
|
"hpsv2.1": get_model_path("HPS_v2/HPS_v2.1_compressed.safetensors"),
|
|
"imagereward": get_model_path("ImageReward/ImageReward.safetensors"),
|
|
"med_config": get_model_path("ImageReward/med_config.json"),
|
|
"clip": get_model_path("CLIP-ViT-H-14-laion2B-s32B-b79K"),
|
|
"clip-large": get_model_path("clip-vit-large-patch14"),
|
|
"mps": get_model_path("MPS_overall_checkpoint/MPS_overall_checkpoint_diffsynth.pth"),
|
|
"pickscore": get_model_path("PickScore_v1")
|
|
} |