dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-20 10:35:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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 != "":