Hugging Face's logo

Crystalcareai
/
Quiet-Star-Custom

Text Generation
Transformers
Safetensors
quiet
custom_code
Model card Files Community
2
Crystalcareai commited on
Commit
5a3b899
·
verified ·
1 Parent(s): def2825

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +1 -1
generate.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from transformers import LogitsProcessorList, StoppingCriteriaList
3
 
4
- def custom_generate(
5
  self,
6
  input_ids,
7
  attention_mask=None,
 
1
  import torch
2
  from transformers import LogitsProcessorList, StoppingCriteriaList
3
 
4
+ def generate(
5
  self,
6
  input_ids,
7
  attention_mask=None,