Spaces:

TeamSAS
/

UB_VSA

Running

SakshamLak commited on Apr 29

Commit

3596f15

verified ·

1 Parent(s): 00908f3

Update buffalo_rag/model/rag.py

Files changed (1) hide show

buffalo_rag/model/rag.py CHANGED Viewed

@@ -129,7 +129,7 @@ Answer:"""
 # Example usage
 if __name__ == "__main__":
-    rag = BuffaloRAG(model_name="Qwen/Qwen1.5-1.8B-Chat")
     response = rag.answer("How do I apply for OPT?")
     print(f"Query: {response['query']}")

 # Example usage
 if __name__ == "__main__":
+    rag = BuffaloRAG(model_name="1bitLLM/bitnet_b1_58-large")
     response = rag.answer("How do I apply for OPT?")
     print(f"Query: {response['query']}")