Spaces:
Sleeping
Sleeping
Daniel Fried
commited on
Commit
·
3d9bc9d
1
Parent(s):
4bb384f
change model
Browse files- 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-
|
31 |
print("loading tokenizer")
|
32 |
-
tokenizer = AutoTokenizer.from_pretrained("facebook/incoder-
|
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:
|