TobDeBer commited on
Commit
c09542b
·
1 Parent(s): eab4117

original model

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -25,8 +25,9 @@ if len(sys.argv) > 1 and os.path.exists(sys.argv[1]):
25
  MODEL_NAME = sys.argv[1]
26
  print(f"Using local model from: {MODEL_NAME}")
27
  else:
28
- MODEL_NAME = "TobDeBer/SmolLM2-135M-Instruct-q99-bnb4" # "HuggingFaceTB/SmolLM2-135M-Instruct"
29
-
 
30
  # Global variables
31
  tokenizer = None
32
  model = None
 
25
  MODEL_NAME = sys.argv[1]
26
  print(f"Using local model from: {MODEL_NAME}")
27
  else:
28
+ #MODEL_NAME = "TobDeBer/SmolLM2-135M-Instruct-q99-bnb4"
29
+ MODEL_NAME = "HuggingFaceTB/SmolLM2-135M-Instruct"
30
+
31
  # Global variables
32
  tokenizer = None
33
  model = None