Update app.py

app.py

@@ -48,7 +48,6 @@ def fix_lora_state_dict(state_dict: dict) -> dict:
 # ══════════════════════════════════════════════════════════════
 RGB_LATENT_SCALE = 0.18215
 
-SD_MODEL_ID  = os.environ.get("SD_MODEL_ID",      "sd-research/stable-diffusion-2-1-base")
 ADAPTOR_REPO = os.environ.get("ADAPTOR_MODEL_ID", "UEXdo/HeightAdaptor-weight")
 
 DATASET_CFG = {
@@ -88,7 +87,7 @@ def build_model(dataset_name: str, h_type: str):
     print(f"🔧 Building model — dataset={dataset_name}, h_type={h_type}")
 
     pipe = StableDiffusionPipeline.from_pretrained(
-        SD_MODEL_ID,
+        os.path.join(ADAPTOR_DIR, "stable-diffusion-v2"),
         torch_dtype=torch.float32,
         safety_checker=None,
         requires_safety_checker=False,