Spaces:

codeparrot
/

codeparrot-generation

Build error

lvwerra HF Staff commited on Dec 6, 2021

Commit

0e10e84

1 Parent(s): 56eec5a

fix device logic

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def load_examples():
 st.set_page_config(page_icon=':parrot:', layout="wide")
-device = 1 if torch.cuda.is_available() else 0
 device_name = "GPU" if torch.cuda.is_available() else "CPU"
 model_ckpt = "lvwerra/codeparrot"
 pipe = load_pipeline(model_ckpt, device)

 st.set_page_config(page_icon=':parrot:', layout="wide")
+device = 0 if torch.cuda.is_available() else -1
 device_name = "GPU" if torch.cuda.is_available() else "CPU"
 model_ckpt = "lvwerra/codeparrot"
 pipe = load_pipeline(model_ckpt, device)