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"
|