change cuda to cpu
Browse files
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('
|
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()
|