Spaces:
Running
on
T4
Running
on
T4
Commit
•
1c60403
1
Parent(s):
8747810
Update vtoonify_model.py
Browse files- vtoonify_model.py +1 -0
vtoonify_model.py
CHANGED
|
@@ -16,6 +16,7 @@ import torch.nn.functional as F
|
|
| 16 |
from torchvision import transforms
|
| 17 |
from model.encoder.align_all_parallel import align_face
|
| 18 |
import gc
|
|
|
|
| 19 |
|
| 20 |
MODEL_REPO = 'PKUWilliamYang/VToonify'
|
| 21 |
|
|
|
|
| 16 |
from torchvision import transforms
|
| 17 |
from model.encoder.align_all_parallel import align_face
|
| 18 |
import gc
|
| 19 |
+
import huggingface_hub
|
| 20 |
|
| 21 |
MODEL_REPO = 'PKUWilliamYang/VToonify'
|
| 22 |
|