Spaces:
Runtime error
Runtime error
Arnaudding001
commited on
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:
|