BlueDice
/

Katakuri-6b-torch

Inference Endpoints

Model card Files Files and versions Community

BlueDice commited on May 17, 2023

Commit

62a14f0

·

1 Parent(s): 78bd8dc

Update code/inference.py

Files changed (1) hide show

code/inference.py +1 -1

code/inference.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
 import re
 import torch

+from transformers import AutoTokenizer
 import re
 import torch