isayahc commited on
Commit
e34519b
·
1 Parent(s): ba58b26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -47,7 +47,8 @@ model_id = "TheBloke/zephyr-7B-beta-GGUF"
47
  print( "initalized model")
48
 
49
  # tokenizer = AutoTokenizer.from_pretrained(model_id)
50
- model = AutoModel.from_pretrained("TheBloke/zephyr-7B-beta-GGUF")
 
51
 
52
 
53
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
47
  print( "initalized model")
48
 
49
  # tokenizer = AutoTokenizer.from_pretrained(model_id)
50
+ model = AutoModelForCausalLM.from_pretrained(model_id)
51
+ # model = AutoModel.from_pretrained("TheBloke/zephyr-7B-beta-GGUF")
52
 
53
 
54
  tokenizer = AutoTokenizer.from_pretrained(model_id)