Converting Safetensor to torch

[maciek226]

I am trying to convert .safetensor file into .pth to use in older code with new weights (Llama 3 8B - R1). Since I was having problems, I tried to convert the weights of LLama 3 8B as the model contains both a .pth file and .safetensor. The model did not work well so I also wrote a script that looked at a row of the original wights and then searched for the row in the converted tensor. Here is the output:

0 0
1 64
2 1
3 65
4 2
5 66
6 3
7 67
8 4
9 68
10 5
11 69
12 6
13 70
14 7
15 71
16 8
17 72
18 9
19 51
19 73
20 10
21 74
22 11
23 75
24 12
25 76
26 13
27 77
28 14
29 78
30 15
31 79
32 16
33 80
34 17
35 81
36 18
36 91
37 82
38 19
39 83
40 20
...

I am getting the same problem when using safe_open and with safetensors.torch.load_file. Does anyone have any idea what could be causing this type of issue?