kadirnar commited on
Commit
ba62c7b
1 Parent(s): 2ef4a4b

Update utils_app.py

Browse files
Files changed (1) hide show
  1. utils_app.py +2 -1
utils_app.py CHANGED
@@ -45,7 +45,8 @@ def keras_stable_diffusion(
45
  num_steps=num_inference_step,
46
  unconditional_guidance_scale=guidance_scale
47
  )
48
- tf.Session().close()
 
49
 
50
  return generated_images
51
 
 
45
  num_steps=num_inference_step,
46
  unconditional_guidance_scale=guidance_scale
47
  )
48
+ tf.keras.backend.clear_session()
49
+
50
 
51
  return generated_images
52