Spaces:
Paused
Paused
Commit
·
c1bb08d
1
Parent(s):
aed3865
Specify checkpoint
Browse files
app.py
CHANGED
|
@@ -3,6 +3,8 @@ import transformers as t
|
|
| 3 |
import torch
|
| 4 |
import peft
|
| 5 |
|
|
|
|
|
|
|
| 6 |
# Load your fine-tuned model and tokenizer
|
| 7 |
tokenizer = t.AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-hf")
|
| 8 |
model = t.AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-hf", load_in_8bit=True, torch_dtype=torch.float16)
|
|
|
|
| 3 |
import torch
|
| 4 |
import peft
|
| 5 |
|
| 6 |
+
checkpoint = 1200
|
| 7 |
+
|
| 8 |
# Load your fine-tuned model and tokenizer
|
| 9 |
tokenizer = t.AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-hf")
|
| 10 |
model = t.AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-hf", load_in_8bit=True, torch_dtype=torch.float16)
|