Spaces:

sschet
/

Vicuna_13B

Paused

sschet commited on May 16, 2023

Commit

e9b4eca

•

1 Parent(s): 7238493

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 @st.cache(allow_output_mutation=True)
 def load_model():
-    model = LlamaForCausalLM.from_pretrained('/main/Weights')
     tokenizer = AutoTokenizer.from_pretrained('/code/Tokenizer/tokenizer.model')
     return model, tokenizer

 @st.cache(allow_output_mutation=True)
 def load_model():
+    model = LlamaForCausalLM.from_pretrained('/Weights')
     tokenizer = AutoTokenizer.from_pretrained('/code/Tokenizer/tokenizer.model')
     return model, tokenizer