dipta007 commited on
Commit
e061b37
·
1 Parent(s): 1e3c1f4

remove load in 8bit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def get_pipeline(model_name):
44
  # chatbot = pipeline(model=model_name, task="text-generation", device=device)
45
 
46
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=TOKEN)
47
- model = AutoModelForCausalLM.from_pretrained(model_name, token=TOKEN, load_in_8bit=True)
48
  # chatbot = pipeline("conversational", model=model, tokenizer=tokenizer, device=device)
49
  chatbot = pipeline("conversational", model=model, tokenizer=tokenizer)
50
  return chatbot
 
44
  # chatbot = pipeline(model=model_name, task="text-generation", device=device)
45
 
46
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=TOKEN)
47
+ model = AutoModelForCausalLM.from_pretrained(model_name, token=TOKEN)
48
  # chatbot = pipeline("conversational", model=model, tokenizer=tokenizer, device=device)
49
  chatbot = pipeline("conversational", model=model, tokenizer=tokenizer)
50
  return chatbot