update model loading

app.py

@@ -14,27 +14,20 @@ from huggingface_hub import login, snapshot_download
 
 def _load_model_processor():
 
-    local_vlm_dir = snapshot_download(
+    local_dir = snapshot_download(
         repo_id="xiabs/DreamOmni2",
         revision="main",
-        allow_patterns=["vlm-model/**"],
-    )
-    local_lora_dir = snapshot_download(
-        repo_id="xiabs/DreamOmni2",
-        revision="main",
-        allow_patterns=["edit_lora/**"],
+        allow_patterns=["vlm-model/**", "edit_lora/**"],
     )
+    local_vlm_dir = os.path.join(local_dir, 'vlm-model')
+    local_lora_dir = os.path.join(local_dir, 'edit_lora')
 
     print(f"Loading models from vlm_path: {local_vlm_dir}, edit_lora_path: {local_lora_dir}")
     pipe = FluxKontextPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-Kontext-dev",
         torch_dtype=torch.bfloat16
     )
-    pipe.load_lora_weights(
-        local_lora_dir,
-        adapter_name="edit",
-        weight_name="pytorch_lora_weights.safetensors"
-    )
+    pipe.load_lora_weights(local_lora_dir, adapter_name="edit")
     pipe.set_adapters(["edit"], adapter_weights=[1])    
 
     vlm_model = Qwen2_5_VLForConditionalGeneration.from_pretrained(