Spaces:

JBHF
/

Literature_Based_Research_1

Sleeping

JBHF commited on Mar 1

Commit

189ffdf

•

1 Parent(s): e65901c

Update rag.py

Files changed (1) hide show

rag.py CHANGED Viewed

@@ -64,7 +64,7 @@ class ChatPDF:
         #
         # https://huggingface.co/stabilityai/stablelm-3b-4e1t?library=true
         tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-3b-4e1t")
-        model = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-3b-4e1t")
         self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1024, chunk_overlap=100)

         #
         # https://huggingface.co/stabilityai/stablelm-3b-4e1t?library=true
         tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-3b-4e1t")
+        self.model = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-3b-4e1t")
         self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1024, chunk_overlap=100)