nisten commited on
Commit
01152d3
1 Parent(s): 1c1b839

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -36,7 +36,7 @@ def start_worker(model_path: str, bits=16):
36
  "-m",
37
  "llava.serve.model_worker",
38
  "--host",
39
- "0.0.0.0",
40
  "--controller",
41
  "http://localhost:10000",
42
  "--model-path",
@@ -53,7 +53,7 @@ def start_worker(model_path: str, bits=16):
53
 
54
  if __name__ == "__main__":
55
  parser = argparse.ArgumentParser()
56
- parser.add_argument("--host", type=str, default="0.0.0.0")
57
  parser.add_argument("--port", type=int)
58
  parser.add_argument("--controller-url", type=str, default="http://localhost:10000")
59
  parser.add_argument("--concurrency-count", type=int, default=5)
@@ -78,7 +78,7 @@ Set the environment variable `model` to change the model:
78
  print(f"args: {gws.args}")
79
 
80
  model_path = os.getenv("model", "liuhaotian/llava-v1.6-mistral-7b")
81
- bits = int(os.getenv("bits", 4))
82
  concurrency_count = int(os.getenv("concurrency_count", 5))
83
 
84
  controller_proc = start_controller()
@@ -95,8 +95,8 @@ Set the environment variable `model` to change the model:
95
  api_open=False
96
  ).launch(
97
  server_name=gws.args.host,
98
- server_port=gws.args.port,
99
- share=gws.args.share
100
  )
101
 
102
  except Exception as e:
 
36
  "-m",
37
  "llava.serve.model_worker",
38
  "--host",
39
+ "ec2-3-239-127-158.compute-1.amazonaws.com",
40
  "--controller",
41
  "http://localhost:10000",
42
  "--model-path",
 
53
 
54
  if __name__ == "__main__":
55
  parser = argparse.ArgumentParser()
56
+ parser.add_argument("--host", type=str, default="ec2-3-239-127-158.compute-1.amazonaws.com")
57
  parser.add_argument("--port", type=int)
58
  parser.add_argument("--controller-url", type=str, default="http://localhost:10000")
59
  parser.add_argument("--concurrency-count", type=int, default=5)
 
78
  print(f"args: {gws.args}")
79
 
80
  model_path = os.getenv("model", "liuhaotian/llava-v1.6-mistral-7b")
81
+ bits = int(os.getenv("bits", 8))
82
  concurrency_count = int(os.getenv("concurrency_count", 5))
83
 
84
  controller_proc = start_controller()
 
95
  api_open=False
96
  ).launch(
97
  server_name=gws.args.host,
98
+ server_port=8080,
99
+ share=True
100
  )
101
 
102
  except Exception as e: