Hugging Face's logo

Spaces:
Yanqing0327
/
Project
Sleeping

App Files Community
Yanqing0327 commited on
Commit
b25cf77
·
verified ·
1 Parent(s): 3881722

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import LlavaProcessor, LlavaForConditionalGeneration
4
  from PIL import Image
5
 
6
  # 加载模型
7
- model_id = "llava-hf/llava-v1.5-7b"
8
  processor = LlavaProcessor.from_pretrained(model_id)
9
  model = LlavaForConditionalGeneration.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16)
10
 
 
4
  from PIL import Image
5
 
6
  # 加载模型
7
+ model_id = "Yanqing0327/LLaVA-project"
8
  processor = LlavaProcessor.from_pretrained(model_id)
9
  model = LlavaForConditionalGeneration.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16)
10