DonImages commited on
Commit
a6568e1
·
verified ·
1 Parent(s): 736177a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
 
3
  # Specify the correct path to your LoRA checkpoint file
4
- lora_path = "./lora_model.pt" # Replace with the correct file name
5
 
6
  # Define the device (use CPU in this case)
7
  device = torch.device("cpu")
 
1
  import torch
2
 
3
  # Specify the correct path to your LoRA checkpoint file
4
+ lora_path = "./lora_trained_model.pt" # Replace with the correct file name
5
 
6
  # Define the device (use CPU in this case)
7
  device = torch.device("cpu")