Update app.py

app.py

@@ -103,8 +103,6 @@ pipe_realpony.unet.set_attn_processor(AttnProcessor2_0())
 pipe_noobai.unet.set_attn_processor(AttnProcessor2_0())
 pipe_hybridpony.unet.set_attn_processor(AttnProcessor2_0())
 
-lora_model_path = "models/lora/LowStepBeta_v001.safetensors"
-lora = load_file(lora_model_path).to("cuda")
 
 # Define samplers
 samplers = {
@@ -128,6 +126,9 @@ DEFAULT_NEGATIVE_SUFFIX = "low quality, low resolution, simple background, bad c
 # Initialize Florence model
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+lora_model_path = "models/lora/LowStepBeta_v001.safetensors"
+lora = load_file(lora_model_path, device=device)
+
 #def load_models():
     #with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
 florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()