Spaces:
Runtime error
Runtime error
Commit
·
30131c2
1
Parent(s):
648ad35
Update encoder_psp.py
Browse files- encoder_psp.py +2 -2
encoder_psp.py
CHANGED
|
@@ -7,8 +7,8 @@ import math
|
|
| 7 |
|
| 8 |
import torch
|
| 9 |
from torch import nn
|
| 10 |
-
from
|
| 11 |
-
from
|
| 12 |
|
| 13 |
def get_keys(d, name):
|
| 14 |
if 'state_dict' in d:
|
|
|
|
| 7 |
|
| 8 |
import torch
|
| 9 |
from torch import nn
|
| 10 |
+
from encoder_encoders import psp_encoders
|
| 11 |
+
from stylegan_model import Generator
|
| 12 |
|
| 13 |
def get_keys(d, name):
|
| 14 |
if 'state_dict' in d:
|