grab .1 GPU
Browse files- 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.
|
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"
|