Update app.py

app.py

@@ -112,8 +112,10 @@ Edward:''', 333, 1, 0.3, 0, 1],
 
 ########################## visual rwkv ################################################################
 visual_title = 'ViusualRWKV-v5'
-rwkv_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_rwkv.pth"
-vision_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_visual.pth"
+#rwkv_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_rwkv.pth"
+#vision_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_visual.pth"
+rwkv_remote_path = "rwkv3b-vitl336p14-577token_mix665k_8gpu_rwkv.pth"
+vision_remote_path = "rwkv3b-vitl336p14-577token_mix665k_8gpu_visual.pth"
 vision_tower_name = 'openai/clip-vit-large-patch14-336'
 
 model_path = hf_hub_download(repo_id="howard-hou/visualrwkv-5", filename=rwkv_remote_path)