Spaces:
Runtime error
Runtime error
Danil
commited on
Commit
Β·
2c3abf7
1
Parent(s):
0bea55f
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ def load_model():
|
|
7 |
Loads the model and tokenizer from the local directory.
|
8 |
:return: A list containing the model and the tokenizer.
|
9 |
'''
|
10 |
-
model_name = 'facebook/incoder-
|
11 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
12 |
model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
|
13 |
return [model, tokenizer]
|
|
|
7 |
Loads the model and tokenizer from the local directory.
|
8 |
:return: A list containing the model and the tokenizer.
|
9 |
'''
|
10 |
+
model_name = 'facebook/incoder-1B'
|
11 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
12 |
model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
|
13 |
return [model, tokenizer]
|