Hugging Face's logo

Spaces:
helmfridsson
/
citrus
Sleeping

App Files Community
tomas.helmfridsson commited on
Commit
cd31712
·
1 Parent(s): 4416daf

LLM_MODEL TinyLlama/TinyLlama-1.1B-Chat-v1.0

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ EMB_MODEL = "KBLab/sentence-bert-swedish-cased"
15
  #LLM_MODEL = "google/flan-t5-base" # Dålig
16
  #LLM_MODEL = "bigscience/bloom-560m" # Dålig
17
  #LLM_MODEL = "NbAiLab/nb-gpt-j-6B" #- Restricted
18
- LLM_MODEL = "datificate/gpt2-small-swedish"
 
19
 
20
  CHUNK_SIZE = 400
21
  CHUNK_OVERLAP = 40
 
15
  #LLM_MODEL = "google/flan-t5-base" # Dålig
16
  #LLM_MODEL = "bigscience/bloom-560m" # Dålig
17
  #LLM_MODEL = "NbAiLab/nb-gpt-j-6B" #- Restricted
18
+ #LLM_MODEL = "datificate/gpt2-small-swedish" # Finns ej på Hugging face
19
+ LLM_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
20
 
21
  CHUNK_SIZE = 400
22
  CHUNK_OVERLAP = 40