zideliu commited on
Commit
815b624
·
1 Parent(s): 27f50b2

Update config

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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):