snorkelai
/

RedPajama-7B-Chat-Curated

Text Generation

text-generation-inference

Model card Files Files and versions

viethoangtranduong commited on Jun 19, 2023

Commit

99189b4

·

1 Parent(s): f2e31d0

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -1,5 +1,5 @@
 # import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:1500' # A10
-import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:1500' # A100
 # import torch

 # import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:1500' # A10
+import os; os.environ["PYTORCH_CUDA_ALLOC_CONF"] = 'max_split_size_mb:2000' # A100
 # import torch