Spaces:
Running
on
Zero
Running
on
Zero
okaris
commited on
Commit
·
50c6839
1
Parent(s):
9c24db9
ZeroGPU hack?
Browse files- omni_zero.py +4 -0
omni_zero.py
CHANGED
|
@@ -7,6 +7,10 @@ sys.path.insert(0, './diffusers/src')
|
|
| 7 |
import torch
|
| 8 |
import torch.nn as nn
|
| 9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
from huggingface_hub import snapshot_download
|
| 11 |
from diffusers import DPMSolverMultistepScheduler
|
| 12 |
from diffusers.models import ControlNetModel
|
|
|
|
| 7 |
import torch
|
| 8 |
import torch.nn as nn
|
| 9 |
|
| 10 |
+
#Hack for ZeroGPU
|
| 11 |
+
torch.jit.script = lambda f: f
|
| 12 |
+
####
|
| 13 |
+
|
| 14 |
from huggingface_hub import snapshot_download
|
| 15 |
from diffusers import DPMSolverMultistepScheduler
|
| 16 |
from diffusers.models import ControlNetModel
|