Spaces:

pmelnechuk
/

QAMath

Runtime error

pmelnechuk commited on 1 day ago

Commit

a74ed47

verified ·

1 Parent(s): cd66bdc

Update src/model_load.py

Files changed (1) hide show

src/model_load.py CHANGED Viewed

@@ -5,7 +5,9 @@ import torch
 from langchain.prompts import PromptTemplate
 from langchain.llms import HuggingFaceHub
 from langchain.chains import LLMChain
 def load_model():
     model_name="tiiuae/Falcon3-7B-Instruct"

 from langchain.prompts import PromptTemplate
 from langchain.llms import HuggingFaceHub
 from langchain.chains import LLMChain
+import spaces
+@spaces.GPU
 def load_model():
     model_name="tiiuae/Falcon3-7B-Instruct"