howard-hou commited on
Commit
463db3d
·
verified ·
1 Parent(s): f5edc96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -123,7 +123,7 @@ visual_rwkv = RWKV(model=model_path, strategy='cuda fp16')
123
 
124
  ##########################################################################
125
  from modeling_vision import VisionEncoder, VisionEncoderConfig
126
- config = VisionEncoderConfig(n_embd=model.args.n_embd,
127
  vision_tower_name=vision_tower_name,
128
  grid_size=-1)
129
  visual_encoder = VisionEncoder(config)
 
123
 
124
  ##########################################################################
125
  from modeling_vision import VisionEncoder, VisionEncoderConfig
126
+ config = VisionEncoderConfig(n_embd=visual_rwkv.args.n_embd,
127
  vision_tower_name=vision_tower_name,
128
  grid_size=-1)
129
  visual_encoder = VisionEncoder(config)