Spaces:
Running on Zero
Running on Zero
Commit ·
a22d9ab
1
Parent(s): f14abb6
test torch 2.8
Browse files- app.py +1 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -113,6 +113,7 @@ def safe_image_input(image):
|
|
| 113 |
|
| 114 |
|
| 115 |
# ====== 推理函数 ======
|
|
|
|
| 116 |
@torch.no_grad()
|
| 117 |
def run_inference(image_ref, image_tgt, do_rm_bkg):
|
| 118 |
image_ref = safe_image_input(image_ref)
|
|
|
|
| 113 |
|
| 114 |
|
| 115 |
# ====== 推理函数 ======
|
| 116 |
+
@spaces.GPU
|
| 117 |
@torch.no_grad()
|
| 118 |
def run_inference(image_ref, image_tgt, do_rm_bkg):
|
| 119 |
image_ref = safe_image_input(image_ref)
|
requirements.txt
CHANGED
|
@@ -13,7 +13,7 @@ pillow
|
|
| 13 |
huggingface_hub>=0.23
|
| 14 |
pytorch-lightning
|
| 15 |
scipy
|
| 16 |
-
torch
|
| 17 |
torchmetrics
|
| 18 |
torchvision
|
| 19 |
tqdm
|
|
|
|
| 13 |
huggingface_hub>=0.23
|
| 14 |
pytorch-lightning
|
| 15 |
scipy
|
| 16 |
+
torch==2.8.0
|
| 17 |
torchmetrics
|
| 18 |
torchvision
|
| 19 |
tqdm
|