Spaces:
Runtime error
Runtime error
Commit
·
ec23ee9
1
Parent(s):
88533e2
change model to Salesforce/codegen2-1B
Browse files
app.py
CHANGED
|
@@ -2,7 +2,8 @@ import gradio as gr
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 3 |
|
| 4 |
# checkpoint = "Salesforce/codegen25-7b-instruct"
|
| 5 |
-
checkpoint = "Salesforce/codegen-2B-nl"
|
|
|
|
| 6 |
|
| 7 |
# tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
|
| 8 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, cache_dir="models/")
|
|
|
|
| 2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 3 |
|
| 4 |
# checkpoint = "Salesforce/codegen25-7b-instruct"
|
| 5 |
+
# checkpoint = "Salesforce/codegen-2B-nl"
|
| 6 |
+
checkpoint = "Salesforce/codegen2-1B"
|
| 7 |
|
| 8 |
# tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
|
| 9 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, cache_dir="models/")
|