jhtonyKoo commited on
Commit
8889a57
1 Parent(s): bb9523a

modify fx norm

Browse files
Files changed (1) hide show
  1. inference.py +1 -0
inference.py CHANGED
@@ -233,6 +233,7 @@ class MasteringStyleTransfer:
233
  if isinstance(param_value, torch.Tensor):
234
  param_value = param_value.item()
235
 
 
236
  if fx_name in param_mapper and param_name in param_mapper[fx_name]:
237
  friendly_name, unit, min_val, max_val = param_mapper[fx_name][param_name]
238
  if fx_name == 'IMAGER' and param_name == 'width':
 
233
  if isinstance(param_value, torch.Tensor):
234
  param_value = param_value.item()
235
 
236
+ print(f"fx name: {fx_name} param_name: {param_name}")
237
  if fx_name in param_mapper and param_name in param_mapper[fx_name]:
238
  friendly_name, unit, min_val, max_val = param_mapper[fx_name][param_name]
239
  if fx_name == 'IMAGER' and param_name == 'width':