Locutusque commited on
Commit
ffbab41
·
verified ·
1 Parent(s): 5882bc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -15,7 +15,6 @@ def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_len
15
  # Generate text using the model
16
  generator.model.cuda()
17
  generator.device = torch.device("cuda")
18
- prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
19
  outputs = generator(
20
  prompt,
21
  do_sample=True,
 
15
  # Generate text using the model
16
  generator.model.cuda()
17
  generator.device = torch.device("cuda")
 
18
  outputs = generator(
19
  prompt,
20
  do_sample=True,