karimbenharrak commited on
Commit
9d60b6e
1 Parent(s): 1fd6255

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -3
handler.py CHANGED
@@ -59,7 +59,7 @@ class EndpointHandler():
59
 
60
  #self.pipe2.enable_xformers_memory_efficient_attention()
61
 
62
- image = pipe(
63
  prompt=prompt,
64
  image=image,
65
  mask_image=mask_image,
@@ -70,10 +70,9 @@ class EndpointHandler():
70
  output_type="latent", # let's keep in latent to save some VRAM
71
  ).images[0]
72
 
73
- pipe = AutoPipelineForImage2Image.from_pipe(pipe)
74
  #self.pipe3.enable_xformers_memory_efficient_attention()
75
 
76
- image = pipe(
77
  prompt=prompt,
78
  image=image,
79
  guidance_scale=8.0,
 
59
 
60
  #self.pipe2.enable_xformers_memory_efficient_attention()
61
 
62
+ image = self.pipe2(
63
  prompt=prompt,
64
  image=image,
65
  mask_image=mask_image,
 
70
  output_type="latent", # let's keep in latent to save some VRAM
71
  ).images[0]
72
 
 
73
  #self.pipe3.enable_xformers_memory_efficient_attention()
74
 
75
+ image = self.pipe3(
76
  prompt=prompt,
77
  image=image,
78
  guidance_scale=8.0,