Update synonymous_logit_processor.py

synonymous_logit_processor.py

@@ -45,14 +45,18 @@ def generate_candidate_codons_with_generate(initial_codons, temperature=1.0, top
     """
     # Use global variables if not provided as parameters
     if model is None:
-        import builtins
-        model = getattr(builtins, 'model', globals().get('model'))
+        import inspect
+        # Check calling frame's globals and locals
+        frame = inspect.currentframe().f_back
+        model = frame.f_locals.get('model') or frame.f_globals.get('model')
         if model is None:
             raise ValueError("Model not provided and no global 'model' variable found")
     
     if tokenizer is None:
-        import builtins
-        tokenizer = getattr(builtins, 'tokenizer', globals().get('tokenizer'))
+        import inspect
+        # Check calling frame's globals and locals  
+        frame = inspect.currentframe().f_back
+        tokenizer = frame.f_locals.get('tokenizer') or frame.f_globals.get('tokenizer')
         if tokenizer is None:
             raise ValueError("Tokenizer not provided and no global 'tokenizer' variable found")