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)
|