ponytail commited on
Commit
2619577
·
verified ·
1 Parent(s): 8afb396

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = 1
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()