alonsosilva commited on
Commit
d674c41
1 Parent(s): ac4e505

Change model to tiny_starcoder_py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ from solara.alias import rv as v
10
  from typing import Any, Callable, Optional, TypeVar, Union, cast, overload
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
 
13
- tokenizer = AutoTokenizer.from_pretrained('replit/replit-code-v1_5-3b', trust_remote_code=True)
14
- model = AutoModelForCausalLM.from_pretrained('replit/replit-code-v1_5-3b', trust_remote_code=True)
15
 
16
  def use_change(el: reacton.core.Element, on_value: Callable[[Any], Any], enabled=True):
17
  """Trigger a callback when a blur events occurs or the enter key is pressed."""
 
10
  from typing import Any, Callable, Optional, TypeVar, Union, cast, overload
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
 
13
+ tokenizer = AutoTokenizer.from_pretrained('bigcode/tiny_starcoder_py')
14
+ model = AutoModelForCausalLM.from_pretrained('bigcode/tiny_starcoder_py')
15
 
16
  def use_change(el: reacton.core.Element, on_value: Callable[[Any], Any], enabled=True):
17
  """Trigger a callback when a blur events occurs or the enter key is pressed."""