Spaces:

sandz7
/

loki

Runtime error

sandz7 commited on May 23, 2024

Commit

9d8fdc0

1 Parent(s): db79da7

added a newline in header

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def llama_generation(input_text: str,
     # Header prompt
     header = '''Your are a helpful AI called amphisbeana.
-    You will help the user, by giving accurate but creative response'''
     input_ids = llama_tokenizer.encode(input_text + header,
                                        return_tensors='pt').to('cuda')

     # Header prompt
     header = '''Your are a helpful AI called amphisbeana.
+    You will help the user, by giving accurate but creative response.\n'''
     input_ids = llama_tokenizer.encode(input_text + header,
                                        return_tensors='pt').to('cuda')