Spaces:

sschet
/

Vicuna_13B

Paused

sschet commited on May 16, 2023

Commit

7238493

1 Parent(s): 99201eb

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('/tree/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('/main/Weights')
     tokenizer = AutoTokenizer.from_pretrained('/code/Tokenizer/tokenizer.model')
     return model, tokenizer