Leonardo Parente commited on
Commit
bc4906f
1 Parent(s): e77464c

remove 8bits

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ vector_store = SupabaseVectorStore(
27
  )
28
 
29
 
30
- model_path = "01-ai/Yi-6B-Chat-8bits"
31
  tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
32
  model = AutoModelForCausalLM.from_pretrained(
33
  model_path,
 
27
  )
28
 
29
 
30
+ model_path = "01-ai/Yi-6B-Chat"
31
  tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
32
  model = AutoModelForCausalLM.from_pretrained(
33
  model_path,