From d9b3a8753717f242d13b451f3340b051ed50b33f Mon Sep 17 00:00:00 2001
From: shacharNimble <123935894+shacharNimble@users.noreply.github.com>
Date: Wed, 17 Jul 2024 11:39:04 +0300
Subject: [PATCH] Update demo_4M_sampler.py

fix bug: attribute error when Demo4MSampler is called with fm_sr=None and mod=mods_list with mods_list only subset of all.
---
 fourm/demo_4M_sampler.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/fourm/demo_4M_sampler.py b/fourm/demo_4M_sampler.py
index 2639974..5879727 100644
--- a/fourm/demo_4M_sampler.py
+++ b/fourm/demo_4M_sampler.py
@@ -258,6 +258,8 @@ def __init__(self,
             self.mods_sr = mods_sr or list(set(fm_sr.encoder_modalities) | set(fm_sr.decoder_modalities))
         else:
             self.sampler_fm_sr = None
+            self.mods_sr = []
+
 
         # Load tokenizers
         self.toks = {}
@@ -537,4 +539,4 @@ def modalities_to_pil(self, mod_dict, use_fixed_plotting_order=False, resize=Non
             plot_name = MODALITY_PLOTTING_NAME_MAP.get(mod_name, mod_name)
             plotted_modalities.append((img_pil, plot_name))
 
-        return plotted_modalities
\ No newline at end of file
+        return plotted_modalities