Views
No views yet
1import predacons
2
3# Load the model and tokenizer
4model_path = "Precacons/Pico-Lamma-3.2-1B-Reasoning-Instruct"
5model = predacons.load_model(model_path)
6tokenizer = predacons.load_tokenizer(model_path)
7
8# Example usage
9chat = [
10 {"role": "user", "content": "A train travelling at a speed of 60 km/hr is stopped in 15 seconds by applying the brakes. Determine its retardation."},
11]
12res = predacons.chat_generate(model = model,
13 sequence = chat,
14 max_length = 5000,
15 tokenizer = tokenizer,
16 trust_remote_code = True,
17 do_sample=True,
18
19 )
20
21print(res)Pico-Lamma-3.2-1B-Reasoning-Instruct model and use it to generate an explanation for a given query, keeping in mind the limitations mentioned above.