Update generate.py
Browse files- generate.py +0 -2
generate.py
CHANGED
|
@@ -165,8 +165,6 @@ def generate(
|
|
| 165 |
self.rm_initialized = True
|
| 166 |
self.original_mode = False
|
| 167 |
|
| 168 |
-
streamer = TextStreamer(self.tokenizer, skip_prompt=False, skip_special_tokens=True)
|
| 169 |
-
|
| 170 |
generated_token_ids, attention_mask = custom_generate(
|
| 171 |
self,
|
| 172 |
input_ids=input_ids,
|
|
|
|
| 165 |
self.rm_initialized = True
|
| 166 |
self.original_mode = False
|
| 167 |
|
|
|
|
|
|
|
| 168 |
generated_token_ids, attention_mask = custom_generate(
|
| 169 |
self,
|
| 170 |
input_ids=input_ids,
|