Add lora on generate
This commit is contained in:
1 parent
6e1b39c7a8
commit
732c761231
9 files changed
+812
-8
No files matched your search
@@ -160,6 +160,16 @@ class ComponentPool:
|
||||
self._free(entry)
|
||||
return freed
|
||||
|
||||
def evict_where(self, predicate) -> int:
|
||||
"""Free every entry whose module satisfies predicate(module)."""
|
||||
with self._lock:
|
||||
victims = [e for e in self.entries.values() if predicate(e.module)]
|
||||
freed = 0
|
||||
for entry in victims:
|
||||
freed += entry.bytes
|
||||
self._free(entry)
|
||||
return freed
|
||||
|
||||
def clear(self):
|
||||
with self._lock:
|
||||
for entry in list(self.entries.values()):
|
||||
|
||||
Reference in new issue
Block a user