ohayonguy
commited on
Commit
·
9bf7550
1
Parent(s):
cee2118
issue with torch version
Browse files- app.py +1 -0
- requirements.txt +4 -3
app.py
CHANGED
|
@@ -79,6 +79,7 @@ def enhance_face(img, face_helper, has_aligned, only_center_face=False, paste_ba
|
|
| 79 |
dummy_x = torch.zeros_like(cropped_face_t)
|
| 80 |
output = generate_reconstructions(pmrf, dummy_x, cropped_face_t, None, 25, device)
|
| 81 |
restored_face = tensor2img(output.squeeze(0), rgb2bgr=True, min_max=(0, 1))
|
|
|
|
| 82 |
except RuntimeError as error:
|
| 83 |
print(f'\tFailed inference for PMRF: {error}.')
|
| 84 |
restored_face = cropped_face
|
|
|
|
| 79 |
dummy_x = torch.zeros_like(cropped_face_t)
|
| 80 |
output = generate_reconstructions(pmrf, dummy_x, cropped_face_t, None, 25, device)
|
| 81 |
restored_face = tensor2img(output.squeeze(0), rgb2bgr=True, min_max=(0, 1))
|
| 82 |
+
print("\tSucceeded PMRF out")
|
| 83 |
except RuntimeError as error:
|
| 84 |
print(f'\tFailed inference for PMRF: {error}.')
|
| 85 |
restored_face = cropped_face
|
requirements.txt
CHANGED
|
@@ -1,10 +1,11 @@
|
|
|
|
|
| 1 |
torch==2.2.2
|
|
|
|
| 2 |
facexlib==0.2.5
|
| 3 |
realesrgan==0.2.5
|
| 4 |
numpy
|
| 5 |
opencv-python
|
| 6 |
-
|
| 7 |
-
pytorch-lightning==2.4.0
|
| 8 |
scipy
|
| 9 |
tqdm
|
| 10 |
lmdb
|
|
@@ -16,7 +17,7 @@ einops
|
|
| 16 |
torch-ema==0.3
|
| 17 |
huggingface_hub==0.24.6
|
| 18 |
-f https://shi-labs.com/natten/wheels/
|
| 19 |
-
natten==0.17.1+
|
| 20 |
wandb
|
| 21 |
timm
|
| 22 |
torchmetrics
|
|
|
|
| 1 |
+
--index-url https://download.pytorch.org/whl/cu118
|
| 2 |
torch==2.2.2
|
| 3 |
+
torchvision==0.17.2
|
| 4 |
facexlib==0.2.5
|
| 5 |
realesrgan==0.2.5
|
| 6 |
numpy
|
| 7 |
opencv-python
|
| 8 |
+
pytorch-lightning==2.3.3
|
|
|
|
| 9 |
scipy
|
| 10 |
tqdm
|
| 11 |
lmdb
|
|
|
|
| 17 |
torch-ema==0.3
|
| 18 |
huggingface_hub==0.24.6
|
| 19 |
-f https://shi-labs.com/natten/wheels/
|
| 20 |
+
natten==0.17.1+torch220cu118
|
| 21 |
wandb
|
| 22 |
timm
|
| 23 |
torchmetrics
|