Spaces:
Build error
Build error
Commit
·
5f20b07
1
Parent(s):
e88ccc8
fix some Comment
Browse files
app.py
CHANGED
|
@@ -486,10 +486,8 @@ def process_reconstruction(
|
|
| 486 |
gc.collect()
|
| 487 |
torch.cuda.empty_cache()
|
| 488 |
|
| 489 |
-
# 设置随机种子
|
| 490 |
seed_all(seed)
|
| 491 |
-
|
| 492 |
-
# 检查CUDA是否可用
|
| 493 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 494 |
if not torch.cuda.is_available():
|
| 495 |
raise ValueError("CUDA is not available. Check your environment.")
|
|
|
|
| 486 |
gc.collect()
|
| 487 |
torch.cuda.empty_cache()
|
| 488 |
|
|
|
|
| 489 |
seed_all(seed)
|
| 490 |
+
|
|
|
|
| 491 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 492 |
if not torch.cuda.is_available():
|
| 493 |
raise ValueError("CUDA is not available. Check your environment.")
|