DonImages commited on
Commit
70b393b
·
verified ·
1 Parent(s): 7c95ae9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Inspecting the keys: This will show you the structure of the checkpoint and help you identify the correct components to load into your model.
2
+
3
+ import torch # Import torch for loading the checkpoint
4
+
5
+ # Set the path to the LoRA checkpoint file
6
+ lora_path = "/path/to/your/lora_checkpoint.pt" # Replace with the correct path
7
+
8
+ # Load the checkpoint from the specified path
9
+ checkpoint = torch.load(lora_path, map_location=device)
10
+
11
+ # Print the keys to see the structure
12
+ print("Checkpoint keys:", checkpoint.keys())
13
+
14
+ # If you want to inspect the values of a specific key
15
+ for key in checkpoint.keys():
16
+ print(f"Key: {key}, Value shape: {checkpoint[key].shape if hasattr(checkpoint[key], 'shape') else 'N/A'}")
17
+
18
+ # You can also print the entire checkpoint if it is not too large
19
+ # print(checkpoint)