Commit
•
1976155
1
Parent(s):
a923e3f
cuda problem
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -81,7 +81,7 @@ class EndpointHandler():
|
|
| 81 |
controlnet=self.controlnet,
|
| 82 |
torch_dtype=dtype,
|
| 83 |
safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
|
| 84 |
-
)
|
| 85 |
|
| 86 |
|
| 87 |
# Define Generator with seed
|
|
|
|
| 81 |
controlnet=self.controlnet,
|
| 82 |
torch_dtype=dtype,
|
| 83 |
safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
|
| 84 |
+
)
|
| 85 |
|
| 86 |
|
| 87 |
# Define Generator with seed
|