Spaces:
Sleeping
Sleeping
Update config
Browse files
app.py
CHANGED
@@ -15,6 +15,9 @@ from libs.muse import MUSE
|
|
15 |
import utils
|
16 |
import numpy as np
|
17 |
from PIL import Image
|
|
|
|
|
|
|
18 |
|
19 |
def get_config():
|
20 |
config = ml_collections.ConfigDict()
|
@@ -110,9 +113,6 @@ def set_seed(seed: int):
|
|
110 |
torch.manual_seed(seed)
|
111 |
torch.cuda.manual_seed_all(seed)
|
112 |
|
113 |
-
def d(**kwargs):
|
114 |
-
"""Helper of creating a config dict."""
|
115 |
-
return ml_collections.ConfigDict(initial_dictionary=kwargs)
|
116 |
|
117 |
|
118 |
def cfg_nnet(x, context, scale=None,lambdaA=None,lambdaB=None):
|
|
|
15 |
import utils
|
16 |
import numpy as np
|
17 |
from PIL import Image
|
18 |
+
def d(**kwargs):
|
19 |
+
"""Helper of creating a config dict."""
|
20 |
+
return ml_collections.ConfigDict(initial_dictionary=kwargs)
|
21 |
|
22 |
def get_config():
|
23 |
config = ml_collections.ConfigDict()
|
|
|
113 |
torch.manual_seed(seed)
|
114 |
torch.cuda.manual_seed_all(seed)
|
115 |
|
|
|
|
|
|
|
116 |
|
117 |
|
118 |
def cfg_nnet(x, context, scale=None,lambdaA=None,lambdaB=None):
|