WillHeld commited on
Commit
f5ef4f1
·
verified ·
1 Parent(s): f6d9766

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import huggingface_hub
15
  import datasets
16
 
17
  # Model setup
18
- checkpoint = "WillHeld/soft-raccoon"
19
  device = "cuda"
20
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
21
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
 
15
  import datasets
16
 
17
  # Model setup
18
+ checkpoint = "marin-community/marin-8b-instruct"
19
  device = "cuda"
20
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
21
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)