We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c313871 commit a4cbd16Copy full SHA for a4cbd16
bayesflow/networks/flow_matching/flow_matching.py
@@ -67,11 +67,11 @@ def __init__(
67
self.use_optimal_transport = use_optimal_transport
68
69
new_integrate_kwargs = FlowMatching.INTEGRATE_DEFAULT_CONFIG.copy()
70
- new_integrate_kwargs.update(integrate_kwargs)
+ new_integrate_kwargs.update(integrate_kwargs or {})
71
self.integrate_kwargs = new_integrate_kwargs
72
73
new_optimal_transport_kwargs = FlowMatching.OPTIMAL_TRANSPORT_DEFAULT_CONFIG.copy()
74
- new_optimal_transport_kwargs.update(optimal_transport_kwargs)
+ new_optimal_transport_kwargs.update(optimal_transport_kwargs or {})
75
self.optimal_transport_kwargs = new_optimal_transport_kwargs
76
77
self.loss_fn = keras.losses.get(loss_fn)
0 commit comments