mirror of
https://github.com/yuruotong1/autoMate.git
synced 2026-03-22 04:57:18 +08:00
delete gpu warning
This commit is contained in:
2
main.py
2
main.py
@@ -4,8 +4,6 @@ import os
|
||||
from util import download_weights
|
||||
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
|
||||
def run():
|
||||
if not torch.cuda.is_available():
|
||||
print("Warning: GPU is not available, we will use CPU, the application may run slower!\nyou computer will very likely heat up!")
|
||||
download_weights.download()
|
||||
app.run()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user