reaperdoesntknow commited on
Commit
5ab964f
·
verified ·
1 Parent(s): d04a7bc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -30,7 +30,7 @@ Design: Gemma MLP/body + Qwen attention/head, projected and aligned to Gemma’s
30
  from transformers import AutoTokenizer, AutoModelForCausalLM
31
  import torch
32
 
33
- model_id = "reaperdoesntknow/Qemma"
34
  tok = AutoTokenizer.from_pretrained(model_id, use_fast=True)
35
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16).eval()
36
 
 
30
  from transformers import AutoTokenizer, AutoModelForCausalLM
31
  import torch
32
 
33
+ model_id = "reaperdoesntknow/Qemma-sft"
34
  tok = AutoTokenizer.from_pretrained(model_id, use_fast=True)
35
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16).eval()
36