Gausson commited on
Commit
69858a5
·
verified ·
1 Parent(s): 4761582

Update custom_generate/generate.py

Browse files
Files changed (1) hide show
  1. custom_generate/generate.py +3 -0
custom_generate/generate.py CHANGED
@@ -117,6 +117,9 @@ def generate(model,
117
  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
118
  # from utils.sep_cache_utils import SepCache
119
 
 
 
 
120
  from ..functions_2_patch import _validate_model_kwargs, llama_atten_forward
121
  from ..monkey_patching_utils import monkey_patching
122
  from ..sep_cache_utils import SepCache
 
117
  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
118
  # from utils.sep_cache_utils import SepCache
119
 
120
+ print(f"__file__: {__file__}")
121
+ print(f"os.path.abspath(__file__): {os.path.abspath(__file__)}")
122
+
123
  from ..functions_2_patch import _validate_model_kwargs, llama_atten_forward
124
  from ..monkey_patching_utils import monkey_patching
125
  from ..sep_cache_utils import SepCache