naveenmittal commited on
Commit
ca21cbc
·
verified ·
1 Parent(s): 6b714b2

Update hf_moondream.py

Browse files

AutoConfig.from_pretrained() with local directory path to this repo fails because it doesn't find layers.py, rope.py and lora.py.

This change fixes that.

Files changed (1) hide show
  1. hf_moondream.py +3 -1
hf_moondream.py CHANGED
@@ -13,7 +13,9 @@ from .vision import *
13
  from .text import *
14
  from .region import *
15
  from .utils import *
16
-
 
 
17
 
18
  def extract_question(text):
19
  prefix = "<image>\n\nQuestion: "
 
13
  from .text import *
14
  from .region import *
15
  from .utils import *
16
+ from .layers import *
17
+ from .lora import *
18
+ from .rope import *
19
 
20
  def extract_question(text):
21
  prefix = "<image>\n\nQuestion: "