clui commited on
Commit
7ef7fc8
·
verified ·
1 Parent(s): 558d777

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ from transformers import AutoTokenizer
35
  # )
36
 
37
  llm = HuggingFaceLLM(
38
- model_name="microsoft/phi-1_5", # Mały model 1.3B
39
- tokenizer=AutoTokenizer.from_pretrained("microsoft/phi-1_5"),
40
  device_map="auto"
41
  # model_kwargs={
42
  # "quantization_config": quantization_config,
 
35
  # )
36
 
37
  llm = HuggingFaceLLM(
38
+ model_name="eryk-mazus/polka-1.1b", # Mały model 1.3B
39
+ tokenizer=AutoTokenizer.from_pretrained("eryk-mazus/polka-1.1b"),
40
  device_map="auto"
41
  # model_kwargs={
42
  # "quantization_config": quantization_config,