Skip to content

Commit e62a0eb

Browse files
committed
fix max_memory vllm problem
1 parent 7861c5e commit e62a0eb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

delphi/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ class RunConfig(Serializable):
192192
"""Number of GPUs to use for explanation and scoring."""
193193

194194
max_memory: float = field(
195-
default=0.9,
195+
default=0.7,
196196
)
197197
"""Fraction of GPU memory to allocate to running explainer model."""
198198

0 commit comments

Comments
 (0)