Support beam search & parallel generation (#7)

2023-03-10 09:58:21 -08:00
parent 04e5acc08e
commit 1a7eb7da61
16 changed files with 660 additions and 161 deletions
--- a/cacheflow/models/init.py
+++ b/cacheflow/models/init.py
@@ -1,8 +1,10 @@
 from cacheflow.models.input_metadata import InputMetadata
 from cacheflow.models.model_utils import get_model
+from cacheflow.models.model_utils import set_seed


 __all__ = [
-    'get_model',
    'InputMetadata',
+    'get_model',
+    'set_seed'
 ]