deepseek-ai
/

DeepSeek-OCR

Image-Text-to-Text

feature-extraction

vision-language

Model card Files Files and versions

weege007 commited on 22 days ago

Commit

21b7933

·

verified ·

1 Parent(s): 0ed09dd

Update modeling_deepseekocr.py

Files changed (1) hide show

modeling_deepseekocr.py +3 -2

modeling_deepseekocr.py CHANGED Viewed

@@ -703,8 +703,9 @@ class DeepseekOCRForCausalLM(DeepseekV2ForCausalLM):
     def infer(self, tokenizer, prompt='', image_file='', output_path = '', base_size=1024, image_size=640, crop_mode=True, test_compress=False, save_results=False, eval_mode=False, streamer=None):
         self.disable_torch_init()
-        os.makedirs(output_path, exist_ok=True)
-        os.makedirs(f'{output_path}/images', exist_ok=True)
         if prompt and image_file:
             conversation = [

     def infer(self, tokenizer, prompt='', image_file='', output_path = '', base_size=1024, image_size=640, crop_mode=True, test_compress=False, save_results=False, eval_mode=False, streamer=None):
         self.disable_torch_init()
+        if len(output_path) > 0 :
+            os.makedirs(output_path, exist_ok=True)
+            os.makedirs(f'{output_path}/images', exist_ok=True)
         if prompt and image_file:
             conversation = [