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()
|