Danil commited on
Commit
2c3abf7
Β·
1 Parent(s): 0bea55f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-6B'
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]