Spaces:

vsrinivas
/

ChatBot_by_SrinivasV

Sleeping

vsrinivas commited on Feb 5, 2024

Commit

08ad6d9

verified ·

1 Parent(s): d3937b1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import gradio as gr
 # checkpoint = "tiiuae/falcon-7b"
 # checkpoint = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 # checkpoint = "gpt2"
-checkpoint = "amazon/FalconLite2"
 model = AutoModelForCausalLM.from_pretrained(
     checkpoint, device_map="auto",

 # checkpoint = "tiiuae/falcon-7b"
 # checkpoint = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 # checkpoint = "gpt2"
+checkpoint = "vsrinivas/FalconLite2"
 model = AutoModelForCausalLM.from_pretrained(
     checkpoint, device_map="auto",