Spaces:
Running
Running
modify fx norm
Browse files- 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':
|