oleksandrfluxon
/

mpt-7b-instruct-evaluate

Text Generation

text-generation-inference

Model card Files Files and versions Community

oleksandrfluxon commited on Jul 25, 2023

Commit

ed817dc

•

1 Parent(s): 44092ff

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -1,5 +1,6 @@
-from torch import cuda
 import transformers
 from accelerate import dispatch_model, infer_auto_device_map
 from accelerate.utils import get_balanced_memory
 from transformers import BitsAndBytesConfig, StoppingCriteria, StoppingCriteriaList

+import torch
 import transformers
+from torch import cuda
 from accelerate import dispatch_model, infer_auto_device_map
 from accelerate.utils import get_balanced_memory
 from transformers import BitsAndBytesConfig, StoppingCriteria, StoppingCriteriaList