Spaces:

sandz7
/

loki

Runtime error

sandz7 commited on May 23, 2024

Commit

e4851f5

1 Parent(s): 678f5ce

reworded header

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def llama_generation(input_text: str,
     # Header prompt
     header = '''A conversation between a curious human and an AI assistant called Amphisbeana.
-    The assistant helps the user by giving accurate but creative response.\n\n'''
     input_ids = llama_tokenizer.encode(header + input_text,
                                        return_tensors='pt').to('cuda')

     # Header prompt
     header = '''A conversation between a curious human and an AI assistant called Amphisbeana.
+    The assistant helps the user by giving accurate and complete responses, it is free to make suggestions.\n\n'''
     input_ids = llama_tokenizer.encode(header + input_text,
                                        return_tensors='pt').to('cuda')