Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
·
40aa755
1
Parent(s):
2b4b74e
debug
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ trainer = Trainer(HF_TOKEN)
|
|
61 |
with gr.Blocks(css='style.css') as demo:
|
62 |
# if SPACE_ID == ORIGINAL_SPACE_ID:
|
63 |
# show_warning(SHARED_UI_WARNING)
|
64 |
-
|
65 |
show_warning(CUDA_NOT_AVAILABLE_WARNING)
|
66 |
elif (not 'T4' in GPU_DATA):
|
67 |
show_warning(INVALID_GPU_WARNING)
|
|
|
61 |
with gr.Blocks(css='style.css') as demo:
|
62 |
# if SPACE_ID == ORIGINAL_SPACE_ID:
|
63 |
# show_warning(SHARED_UI_WARNING)
|
64 |
+
if not torch.cuda.is_available():
|
65 |
show_warning(CUDA_NOT_AVAILABLE_WARNING)
|
66 |
elif (not 'T4' in GPU_DATA):
|
67 |
show_warning(INVALID_GPU_WARNING)
|