Upload folder using huggingface_hub

custom_generate/generate.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import torch
-from transformers.cache_utils import DynamicCache
+from transformers import DynamicCache, GenerationConfig
 from typing import Any, Dict, List, Optional, Tuple