<fix> fix a model loading bug.

app.py

@@ -103,8 +103,6 @@ def process_image_and_text(condition_image, target_prompt, condition_image_promp
     global pipe, current_task, transformer
     if current_task != task:
         if current_task is None:
-            current_task = task
-
             # insert LoRA
             lora_config = LoraConfig(
                 r=16,
@@ -130,6 +128,8 @@ def process_image_and_text(condition_image, target_prompt, condition_image_promp
                 if isinstance(m, peft.tuners.lora.layer.Linear):
                     m.forward = restore_forward(m)
 
+        current_task = task
+
         # hack LoRA forward
         def create_hacked_forward(module):
             if not hasattr(module, 'original_forward'):