Spaces:

DonImages
/

Testing2

Runtime error

DonImages commited on Feb 1

Commit

265ed58

verified ·

1 Parent(s): 3585972

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ pipeline = StableDiffusion3Pipeline.from_pretrained(
 # Load the LoRA trained weights
 lora_path = "lora_trained_model.pt"  # Ensure this file is uploaded in the Space
 if os.path.exists(lora_path):
-    lora_state_dict = torch.load(lora_path, map_location=device)
     pipeline.load_lora_weights(lora_state_dict)
     print("✅ LoRA weights loaded successfully!")
 else:

 # Load the LoRA trained weights
 lora_path = "lora_trained_model.pt"  # Ensure this file is uploaded in the Space
 if os.path.exists(lora_path):
+    lora_state_dict = torch.load(lora_path, map_location=device, weights_only=True)
     pipeline.load_lora_weights(lora_state_dict)
     print("✅ LoRA weights loaded successfully!")
 else: