sohojoe commited on
Commit
cd4b360
1 Parent(s): 334dcac

grab .1 GPU

Browse files
Files changed (1) hide show
  1. experimental/clip_app.py +1 -1
experimental/clip_app.py CHANGED
@@ -12,7 +12,7 @@ from clip_retrieval.load_clip import load_clip, get_tokenizer
12
 
13
  # @serve.deployment(num_replicas=2, ray_actor_options={"num_cpus": 0.2, "num_gpus": 0.2})
14
  # @serve.deployment(num_replicas=1, ray_actor_options={"num_cpus": 0.2, "num_gpus": 0.0})
15
- @serve.deployment(num_replicas=10, ray_actor_options={"num_cpus": .2, "num_gpus": 0.0})
16
  class CLIPTransform:
17
  def __init__(self):
18
  # os.environ["OMP_NUM_THREADS"] = "20"
 
12
 
13
  # @serve.deployment(num_replicas=2, ray_actor_options={"num_cpus": 0.2, "num_gpus": 0.2})
14
  # @serve.deployment(num_replicas=1, ray_actor_options={"num_cpus": 0.2, "num_gpus": 0.0})
15
+ @serve.deployment(num_replicas=10, ray_actor_options={"num_cpus": .2, "num_gpus": 0.1})
16
  class CLIPTransform:
17
  def __init__(self):
18
  # os.environ["OMP_NUM_THREADS"] = "20"