remove load in 8bit
Browse files
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
|
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
|