Daniel Fried commited on
Commit
3d9bc9d
1 Parent(s): 4bb384f

change model

Browse files
Files changed (1) hide show
  1. modules/app.py +4 -3
modules/app.py CHANGED
@@ -15,7 +15,8 @@ import json
15
  # })
16
  # origins=[f"http://localhost:{PORT}", "https://huggingface.co", "https://hf.space"]
17
 
18
- CUDA = True
 
19
  PORT = 7860
20
  VERBOSE = False
21
 
@@ -27,9 +28,9 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
27
 
28
 
29
  print("loading model")
30
- model = AutoModelForCausalLM.from_pretrained("facebook/incoder-6B")
31
  print("loading tokenizer")
32
- tokenizer = AutoTokenizer.from_pretrained("facebook/incoder-6B")
33
  print("loading complete")
34
 
35
  if CUDA:
15
  # })
16
  # origins=[f"http://localhost:{PORT}", "https://huggingface.co", "https://hf.space"]
17
 
18
+ #CUDA = True
19
+ CUDA = False
20
  PORT = 7860
21
  VERBOSE = False
22
 
28
 
29
 
30
  print("loading model")
31
+ model = AutoModelForCausalLM.from_pretrained("facebook/incoder-1B")
32
  print("loading tokenizer")
33
+ tokenizer = AutoTokenizer.from_pretrained("facebook/incoder-1B")
34
  print("loading complete")
35
 
36
  if CUDA: