Spaces:

sonalkum
/

GAMA

Running on Zero

sonalkum commited on Jul 4, 2024

Commit

7cbf887

1 Parent(s): ef25209

app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ import time,json
 device = "cuda" if torch.cuda.is_available() else "cpu"
-base_model = "/fs/nexus-projects/brain_project/Llama-2-7b-chat-hf-qformer/"
 prompter = Prompter('alpaca_short')
 tokenizer = LlamaTokenizer.from_pretrained(base_model)

 device = "cuda" if torch.cuda.is_available() else "cpu"
+base_model = "./Llama-2-7b-chat-hf-qformer"
 prompter = Prompter('alpaca_short')
 tokenizer = LlamaTokenizer.from_pretrained(base_model)