From c5b3a6f88b350f78c2806feb1d8f17b80fd0a13f Mon Sep 17 00:00:00 2001 From: eliehf Date: Tue, 5 Nov 2024 10:25:12 +0100 Subject: [PATCH] add constants.py file --- its-a-smol-world/src/constants.py | 3 +++ 1 file changed, 3 insertions(+) create mode 100644 its-a-smol-world/src/constants.py diff --git a/its-a-smol-world/src/constants.py b/its-a-smol-world/src/constants.py new file mode 100644 index 0000000..0284ca2 --- /dev/null +++ b/its-a-smol-world/src/constants.py @@ -0,0 +1,3 @@ +MODEL_NAME = "HuggingFaceTB/SmolLM2-1.7B-Instruct" +DEVICE = "mps" +T_TYPE = "bfloat16"