Spaces:
Running
on
Zero
Running
on
Zero
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.")
|