manbeast3b commited on
Commit
d82e44a
·
verified ·
1 Parent(s): 641932b

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +2 -2
src/pipeline.py CHANGED
@@ -32,7 +32,7 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
32
 
33
  for _ in range(1):
34
  deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
35
- # pipeline.scheduler.prepare_loss()
36
  for _ in range(2):
37
  pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
38
  return pipeline
@@ -49,7 +49,7 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
49
  width=request.width,
50
  height=request.height,
51
  generator=generator,
52
- num_inference_steps=20,
53
  cache_interval=1,
54
  cache_layer_id=1,
55
  cache_block_id=0,
 
32
 
33
  for _ in range(1):
34
  deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
35
+ pipeline.scheduler.prepare_loss()
36
  for _ in range(2):
37
  pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
38
  return pipeline
 
49
  width=request.width,
50
  height=request.height,
51
  generator=generator,
52
+ num_inference_steps=13,
53
  cache_interval=1,
54
  cache_layer_id=1,
55
  cache_block_id=0,