Spaces:

DonImages
/

Testing2

Runtime error

DonImages commited on Feb 1

Commit

ada99f9

verified ·

1 Parent(s): 70374ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import torch
 try:
-    lora_checkpoint = torch.load("lora_checkpoint.pt", map_location="cpu")
     print("Checkpoint loaded successfully!")
     print(type(lora_checkpoint))  # Print type to verify structure
 except Exception as e:

 import torch
 try:
+    lora_checkpoint = torch.load("lora_trained_model.pt", map_location="cpu")
     print("Checkpoint loaded successfully!")
     print(type(lora_checkpoint))  # Print type to verify structure
 except Exception as e: