Spaces:
Sleeping
Sleeping
Fix Syntax Error
Browse files
main.py
CHANGED
@@ -65,7 +65,7 @@ async def generate_text(item: Item):
|
|
65 |
|
66 |
# logging.info("Response generated")
|
67 |
|
68 |
-
inp =f"
|
69 |
inputs = tokenizer(inp, return_tensors="pt", padding=True, return_attention_mask=True ).to(model.device)
|
70 |
|
71 |
|
|
|
65 |
|
66 |
# logging.info("Response generated")
|
67 |
|
68 |
+
inp =f"<|system|>You are a helpful assistant<|end|><|user|> {item.prompt} <|end|><|assistant|>"
|
69 |
inputs = tokenizer(inp, return_tensors="pt", padding=True, return_attention_mask=True ).to(model.device)
|
70 |
|
71 |
|