Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load base model
5base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")
6
7# Load LoRA adapters
8model = PeftModel.from_pretrained(base_model, "YOUR_USERNAME/llama3-space-reasoning")
9
10# Load tokenizer
11tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
12
13# Generate
14prompt = "Calculate delta-v for Hohmann transfer from 400 km to 800 km altitude"
15inputs = tokenizer(prompt, return_tensors="pt")
16outputs = model.generate(**inputs, max_new_tokens=512)
17print(tokenizer.decode(outputs[0]))
18
19Example Output
20
21Prompt: Calculate delta-v for a Hohmann transfer from LEO (400 km) to GEO (35,786 km)
22
23Output:
24Step 1: Current orbital velocity
25v₁ = √(μ/r₁) = √(398600/(6371+400)) = 7.67 km/s
26
27Step 2: Transfer orbit semi-major axis
28a = (r₁ + r₂)/2 = (6771 + 42164)/2 = 24467.5 km
29...
30
31Training Code
32
33See https://github.com/isaacasamoah/voyager for full training pipeline.