Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from datetime import datetime
|
|
9 |
import numpy as np
|
10 |
import os
|
11 |
|
12 |
-
os.environ["no_proxy"] = "localhost,127.0.0.1,::1"
|
13 |
|
14 |
|
15 |
def array_to_image_path(image_array):
|
@@ -29,7 +29,7 @@ def array_to_image_path(image_array):
|
|
29 |
return full_path
|
30 |
|
31 |
|
32 |
-
cuda =
|
33 |
model_id = "huangfx1020/human_llama3_8b"
|
34 |
models = {
|
35 |
"HumanLlaVA-8B": LlavaForConditionalGeneration.from_pretrained("huangfx1020/human_llama3_8b", torch_dtype=torch.float16, low_cpu_mem_usage=True ).to(cuda).eval()
|
|
|
9 |
import numpy as np
|
10 |
import os
|
11 |
|
12 |
+
#os.environ["no_proxy"] = "localhost,127.0.0.1,::1"
|
13 |
|
14 |
|
15 |
def array_to_image_path(image_array):
|
|
|
29 |
return full_path
|
30 |
|
31 |
|
32 |
+
cuda = "cpu"
|
33 |
model_id = "huangfx1020/human_llama3_8b"
|
34 |
models = {
|
35 |
"HumanLlaVA-8B": LlavaForConditionalGeneration.from_pretrained("huangfx1020/human_llama3_8b", torch_dtype=torch.float16, low_cpu_mem_usage=True ).to(cuda).eval()
|