Update app.py

app.py

@@ -56,14 +56,16 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
     lora_path = "checkpoints/"
     if orbit_type == "Left":
         weight_name = "orbit_left_lora_weights.safetensors"
-        adapter_name = "orbit_left_lora_weights"
+        #adapter_name = "orbit_left_lora_weights"
     elif orbit_type == "Up":
         weight_name = "orbit_up_lora_weights.safetensors"
-        adapter_name = "orbit_lup_lora_weights"
+        #adapter_name = "orbit_up_lora_weights"
     lora_rank = 256
 
+    adapter_timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+
     # Load LoRA weights on CPU, move to GPU afterward
-    pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=adapter_name)
+    pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=adapter_timestamp)
     pipe.fuse_lora(lora_scale=1 / lora_rank)
     
     # Move the pipeline to GPU for inference
@@ -84,6 +86,7 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
         generator=torch.Generator(device="cpu").manual_seed(seed)
     )
 
+    torch.cuda.empty_cache()
     pipe.unfuse_lora()
     pipe.unload_lora_weights()