daniellefranca96 commited on
Commit
ab1b586
1 Parent(s): 2da383c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -38,7 +38,7 @@ llms = {
38
  }
39
  }
40
 
41
- model = llms[llms.keys()[0]]
42
  llm = Llama(model_path="./code/"+model['file'], n_ctx=nctx, verbose=True, n_threads=8)
43
 
44
 
 
38
  }
39
  }
40
 
41
+ model = llms["TinyLLama 1b OpenOrca 4_K_M 2048"]
42
  llm = Llama(model_path="./code/"+model['file'], n_ctx=nctx, verbose=True, n_threads=8)
43
 
44