Tu Bui
commited on
Commit
•
e97a15c
1
Parent(s):
04acf84
SECRETLEN set by config file
Browse files- Embed_Secret.py +3 -1
Embed_Secret.py
CHANGED
|
@@ -110,7 +110,9 @@ def load_UNet(args):
|
|
| 110 |
|
| 111 |
config = OmegaConf.load(config_file).model
|
| 112 |
secret_len = config.params.secret_len
|
| 113 |
-
|
|
|
|
|
|
|
| 114 |
model = instantiate_from_config(config)
|
| 115 |
state_dict = torch.load(weight_file, map_location=torch.device('cpu'))
|
| 116 |
if 'global_step' in state_dict:
|
|
|
|
| 110 |
|
| 111 |
config = OmegaConf.load(config_file).model
|
| 112 |
secret_len = config.params.secret_len
|
| 113 |
+
global SECRET_LEN
|
| 114 |
+
SECRET_LEN = secret_len
|
| 115 |
+
print(f'Secret length: {secret_len}')
|
| 116 |
model = instantiate_from_config(config)
|
| 117 |
state_dict = torch.load(weight_file, map_location=torch.device('cpu'))
|
| 118 |
if 'global_step' in state_dict:
|