Spaces:
Runtime error
Runtime error
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:
|