Hugging Face's logo

Spaces:
Pusheen
/
LoCo
Runtime error

App Files Community
1
Pusheen commited on
Commit
f3bf446
·
verified ·
1 Parent(s): 10f33f5

Update gligen/ldm/models/diffusion/plms.py

Browse files
Files changed (1) hide show
  1. gligen/ldm/models/diffusion/plms.py +3 -0
gligen/ldm/models/diffusion/plms.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import torch
2
  import numpy as np
3
  from tqdm import tqdm
 
1
+ import os
2
+ os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "0"
3
+ os.environ["CUDA_LAUNCH_BLOCKING"] = "1"
4
  import torch
5
  import numpy as np
6
  from tqdm import tqdm