Hugging Face's logo

Spaces:
EpistemeAI
/
metatune-20b
Sleeping

App Files Community
legolasyiu commited on
Commit
7b98df3
·
verified ·
1 Parent(s): 4c1a418

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import gradio as gr
4
 
5
- checkpoint = "EpistemeAI/metatune-20b"
6
  device = "cpu" # "cuda" or "cpu"
7
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
8
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import gradio as gr
4
 
5
+ checkpoint = "EpistemeAI/metatune-gpt20b-R0"
6
  device = "cpu" # "cuda" or "cpu"
7
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
8
  model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)