Yixin1234 commited on
Commit
f9f96cf
·
1 Parent(s): 59c9715

change cuda to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def describe_image(image, user_question="Describe this image in great detail."):
43
  conversations=conversation,
44
  images=pil_images,
45
  force_batchify=True
46
- ).to('cuda')
47
 
48
  # Load and prepare the model
49
  vl_gpt = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True).to(torch.bfloat16).cuda().eval()
 
43
  conversations=conversation,
44
  images=pil_images,
45
  force_batchify=True
46
+ ).to('cpu')
47
 
48
  # Load and prepare the model
49
  vl_gpt = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True).to(torch.bfloat16).cuda().eval()