path

app.py

@@ -15,6 +15,8 @@ from michelangelo.utils.visualizers import html_util
 
 import gradio as gr
 
+from omegaconf import OmegaConf
+
 from huggingface_hub import snapshot_download
 
 gradio_cached_dir = "./gradio_cached_dir"
@@ -113,7 +115,15 @@ def load_model(model_name: str, model_config_dict: dict, inference_model: Infere
 
         config_ckpt_path = model_config_dict[model_name]
 
-        model_config = get_config_from_file(config_ckpt_path["config"])
+        raw_config_file = config_ckpt_path["config"]
+        raw_config = OmegaConf.load(raw_config_file)
+        raw_clip_ckpt_path = raw_config['model']['params']['first_stage_config']['params']['aligned_module_cfg']['params']['clip_model_version'] 
+        clip_ckpt_path = os.path.join(model_path, raw_clip_ckpt_path)
+        raw_config['model']['params']['first_stage_config']['params']['aligned_module_cfg']['params']['clip_model_version'] = clip_ckpt_path
+        raw_config['model']['params']['cond_stage_config']['params']['version'] = clip_ckpt_path
+        OmegaConf.save(raw_config, 'current_config.yaml')
+
+        model_config = get_config_from_file('current_config.yaml')
         if hasattr(model_config, "model"):
             model_config = model_config.model