Spaces:

howard-hou
/

VisualRWKV-Gradio-1

Sleeping

howard-hou commited on Dec 29, 2023

Commit

ced196e

•

1 Parent(s): f0c200a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ from rwkv.utils import PIPELINE, PIPELINE_ARGS
 pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
 ##########################################################################
-from .model import VisualEncoder, EmbeddingMixer, VisualEncoderConfig
 emb_mixer = EmbeddingMixer(model.w["emb.weight"], num_image_embeddings=4096)
 config = VisualEncoderConfig(n_embd=model.args.n_embd,
  vision_tower_name='openai/clip-vit-large-patch14-336',

 pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
 ##########################################################################
+from modeling import VisualEncoder, EmbeddingMixer, VisualEncoderConfig
 emb_mixer = EmbeddingMixer(model.w["emb.weight"], num_image_embeddings=4096)
 config = VisualEncoderConfig(n_embd=model.args.n_embd,
  vision_tower_name='openai/clip-vit-large-patch14-336',