Commit Graph

15 Commits

Author SHA1 Message Date
de48e6fd59 Putting back main. 2023-06-29 12:08:35 +00:00
0958c588f6 Putting back seed. 2023-06-29 12:07:21 +00:00
c5e8f788be Revert some changes. 2023-06-29 12:05:53 +00:00
e63ed6aaa3 Remove unwrap. 2023-06-29 12:04:25 +00:00
2fe1d3e36d Moving llama to f16. 2023-06-29 12:00:16 +00:00
b4dc9f6108 Add a seed parameter to llama. 2023-06-29 12:47:19 +01:00
1913512f42 Simple example fix. 2023-06-29 11:10:57 +00:00
3872dc4751 Merge pull request #19 from LaurentMazare/llama_safetensors
Llama safetensors
2023-06-29 12:49:26 +02:00
122e334d0c Simplify the pattern matching logic in the cuda backend. 2023-06-29 09:21:11 +01:00
ece3ec6167 Final updates -> moving to deterministic for easier comparison. 2023-06-28 14:56:39 +00:00
926fffa0b7 Ok. 2023-06-28 14:56:39 +00:00
e29dae044d Tmp. 2023-06-28 14:56:38 +00:00
c44e5346f4 Add some helper functions. 2023-06-27 17:37:09 +01:00
318503cd38 Cache the causal mask in llama. 2023-06-27 12:21:08 +01:00
d7f729fb8f Refactor the hierarchy. 2023-06-27 11:57:27 +02:00