fffiloni commited on
Commit
854b688
1 Parent(s): 4583d1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,9 @@ import math
8
 
9
  from huggingface_hub import snapshot_download
10
 
 
 
 
11
  model_ids = [
12
  'runwayml/stable-diffusion-v1-5',
13
  'lllyasviel/sd-controlnet-depth',
 
8
 
9
  from huggingface_hub import snapshot_download
10
 
11
+ os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
12
+
13
+
14
  model_ids = [
15
  'runwayml/stable-diffusion-v1-5',
16
  'lllyasviel/sd-controlnet-depth',