Update README.md
Browse files
README.md
CHANGED
@@ -38,17 +38,12 @@ import requests
|
|
38 |
from PIL import Image
|
39 |
|
40 |
import torch
|
41 |
-
from transformers import AutoProcessor,
|
42 |
|
43 |
|
44 |
model_id = "OpenFace-CQUPT/Human_LLaVA"
|
45 |
cuda = 0
|
46 |
-
model =
|
47 |
-
model_id,
|
48 |
-
torch_dtype=torch.float16,
|
49 |
-
low_cpu_mem_usage=True,
|
50 |
-
trust_remote_code=True
|
51 |
-
).to(cuda)
|
52 |
|
53 |
processor = AutoProcessor.from_pretrained(model_id,trust_remote_code=True)
|
54 |
|
|
|
38 |
from PIL import Image
|
39 |
|
40 |
import torch
|
41 |
+
from transformers import AutoProcessor, AutoModelForPreTraining
|
42 |
|
43 |
|
44 |
model_id = "OpenFace-CQUPT/Human_LLaVA"
|
45 |
cuda = 0
|
46 |
+
model = AutoModelForPreTraining.from_pretrained("OpenFace-CQUPT/Human_LLaVA", torch_dtype=torch.float16).to(cuda)
|
|
|
|
|
|
|
|
|
|
|
47 |
|
48 |
processor = AutoProcessor.from_pretrained(model_id,trust_remote_code=True)
|
49 |
|