ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-15 13:38:07 +00:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

tests: add diffusers tests (#1419)

Browse Source
This commit is contained in:
Ettore Di Giacinto
2023-12-11 02:20:34 -05:00
committed by GitHub
parent 831418612b
commit 48e5380e45
8 changed files with 148 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/python/diffusers/backend_diffusers.py
View File

@ -303,12 +303,17 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
prompt = request.positive_prompt
steps = 1
if request.step != 0:
steps = request.step
# create a dictionary of values for the parameters
options = {
"negative_prompt": request.negative_prompt,
"width": request.width,
"height": request.height,
"num_inference_steps": request.step,
"num_inference_steps": steps,
}
if request.src != "":