update model loading

app.py

@@ -30,7 +30,11 @@ def _load_model_processor():
         "black-forest-labs/FLUX.1-Kontext-dev",
         torch_dtype=torch.bfloat16
     )
-    pipe.load_lora_weights(local_lora_dir, adapter_name="edit")
+    pipe.load_lora_weights(
+        local_lora_dir,
+        adapter_name="edit",
+        weight_name="pytorch_lora_weights.safetensors"
+    )
     pipe.set_adapters(["edit"], adapter_weights=[1])    
 
     vlm_model = Qwen2_5_VLForConditionalGeneration.from_pretrained(

requirements.txt

@@ -12,4 +12,5 @@ torch==2.8.0
 torchaudio==2.8.0
 torchvision==0.23.0
 gradio==5.47.2
-sentencepiece
+sentencepiece
+safetensors