Spaces:
Runtime error
Runtime error
Commit ·
79c54b5
1
Parent(s): c77151e
fix pipe
Browse files- sd_model.py +1 -0
sd_model.py
CHANGED
|
@@ -5,6 +5,7 @@ import spaces
|
|
| 5 |
|
| 6 |
device = "cuda"
|
| 7 |
|
|
|
|
| 8 |
def get_cn_pipeline():
|
| 9 |
controlnets = [
|
| 10 |
ControlNetModel.from_pretrained("./controlnet/lineart", torch_dtype=torch.float16, use_safetensors=True),
|
|
|
|
| 5 |
|
| 6 |
device = "cuda"
|
| 7 |
|
| 8 |
+
@spaces.GPU
|
| 9 |
def get_cn_pipeline():
|
| 10 |
controlnets = [
|
| 11 |
ControlNetModel.from_pretrained("./controlnet/lineart", torch_dtype=torch.float16, use_safetensors=True),
|