Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@
|
|
3 |
import torch # Import torch for loading the checkpoint
|
4 |
|
5 |
# Set the path to the LoRA checkpoint file
|
6 |
-
lora_path = "
|
7 |
|
8 |
# Load the checkpoint from the specified path
|
9 |
checkpoint = torch.load(lora_path, map_location=device)
|
|
|
3 |
import torch # Import torch for loading the checkpoint
|
4 |
|
5 |
# Set the path to the LoRA checkpoint file
|
6 |
+
lora_path = "./lora_checkpoint.pt" # Assuming the checkpoint is in the current folder
|
7 |
|
8 |
# Load the checkpoint from the specified path
|
9 |
checkpoint = torch.load(lora_path, map_location=device)
|