Spaces:

Benson
/

text-generation

Paused

Benson commited on Jul 17, 2023

Commit

32e14fa

1 Parent(s): 26099c5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,8 @@ import torch
 # Create a new FastAPI app instance
 app = FastAPI(docs_url="/")
-gpu = torch.device("cuda")
 # Initialize the text generation pipeline

 # Create a new FastAPI app instance
 app = FastAPI(docs_url="/")
+#gpu = torch.device("cuda")
+gpu = 0 if torch.cuda.is_available() else -1
 # Initialize the text generation pipeline