Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
|
|
92 |
time.sleep(2)
|
93 |
return "", chat_history
|
94 |
'''
|
95 |
-
def respond(message, chat_history, base_model = "/
|
96 |
# define the model and tokenizer.
|
97 |
# model = PhiForCausalLM.from_pretrained(base_model)
|
98 |
model = AutoModelForCausalLM.from_pretrained(base_model)
|
|
|
92 |
time.sleep(2)
|
93 |
return "", chat_history
|
94 |
'''
|
95 |
+
def respond(message, chat_history, base_model = "microsoft/phi-2", device=device): # "meta-llama/Meta-Llama-3-70B"
|
96 |
# define the model and tokenizer.
|
97 |
# model = PhiForCausalLM.from_pretrained(base_model)
|
98 |
model = AutoModelForCausalLM.from_pretrained(base_model)
|