Update README.md
Browse files
README.md
CHANGED
@@ -51,11 +51,11 @@ from qwen_vl_utils import process_vision_info
|
|
51 |
import torch
|
52 |
|
53 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
54 |
-
"2Vasabi/tvl-mini-
|
55 |
)
|
56 |
|
57 |
|
58 |
-
processor = AutoProcessor.from_pretrained("2Vasabi/tvl-mini-
|
59 |
messages = [
|
60 |
{
|
61 |
"role": "user",
|
|
|
51 |
import torch
|
52 |
|
53 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
54 |
+
"2Vasabi/tvl-mini-0.1", torch_dtype=torch.bfloat16, device_map="auto"
|
55 |
)
|
56 |
|
57 |
|
58 |
+
processor = AutoProcessor.from_pretrained("2Vasabi/tvl-mini-0.1")
|
59 |
messages = [
|
60 |
{
|
61 |
"role": "user",
|