[Bugfix] Fix LoRA Scaling on Active Adapters (#1421)
Signed-off-by: Alex Brooks <albrooks@redhat.com>
This commit is contained in:
1 parent
ed08a216c1
commit
bc26de834e
4 files changed
+242
-54
No files matched your search
@@ -192,8 +192,10 @@ class DiffusionWorker:
|
||||
def remove_lora(self, adapter_id: int) -> bool:
|
||||
return self.lora_manager.remove_adapter(adapter_id)
|
||||
|
||||
def add_lora(self, lora_request: LoRARequest, lora_scale: float = 1.0) -> bool:
|
||||
return self.lora_manager.add_adapter(lora_request, lora_scale)
|
||||
def add_lora(self, lora_request: LoRARequest) -> bool:
|
||||
# NOTE (Alex): We have not implemented the API routing
|
||||
# for the frontend server yet.
|
||||
return self.lora_manager.add_adapter(lora_request)
|
||||
|
||||
def list_loras(self) -> list[int]:
|
||||
return self.lora_manager.list_adapters()
|
||||
|
||||
Reference in new issue
Block a user