Skip to content
This repository has been archived by the owner on Jan 9, 2025. It is now read-only.

Commit

Permalink
chore: update resource reservation
Browse files Browse the repository at this point in the history
  • Loading branch information
heiruwu committed Nov 25, 2023
1 parent fc086e7 commit 5c7f0d3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions yolov7/1/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -446,8 +446,8 @@ def undeploy_model(model_name: str):

ray.init(address=model_config.ray_addr)

model_config.ray_actor_options["num_cpus"] = 1
model_config.ray_actor_options["num_gpus"] = 0.2
model_config.ray_actor_options["num_cpus"] = 0.4
model_config.ray_actor_options["num_gpus"] = 0.1

if func == "deploy":
deploy_model(model_config=model_config)
Expand Down

0 comments on commit 5c7f0d3

Please sign in to comment.