Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -28,9 +28,9 @@ from PIL import Image
|
|
| 28 |
import torch
|
| 29 |
|
| 30 |
# Load model and processor
|
| 31 |
-
processor = AutoProcessor.from_pretrained("
|
| 32 |
model = AutoModelForVision2Seq.from_pretrained(
|
| 33 |
-
"
|
| 34 |
torch_dtype=torch.bfloat16,
|
| 35 |
trust_remote_code=True
|
| 36 |
).to("cuda")
|
|
|
|
| 28 |
import torch
|
| 29 |
|
| 30 |
# Load model and processor
|
| 31 |
+
processor = AutoProcessor.from_pretrained("kavinrajkrupsurge/lampe-sim-data-openvla", trust_remote_code=True)
|
| 32 |
model = AutoModelForVision2Seq.from_pretrained(
|
| 33 |
+
"kavinrajkrupsurge/lampe-sim-data-openvla",
|
| 34 |
torch_dtype=torch.bfloat16,
|
| 35 |
trust_remote_code=True
|
| 36 |
).to("cuda")
|