Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,6 @@ device = th.device('cpu' if not th.cuda.is_available() else 'cuda')
|
|
| 26 |
# init stable diffusion model
|
| 27 |
pipe = ComposableStableDiffusionPipeline.from_pretrained(
|
| 28 |
"CompVis/stable-diffusion-v1-4",
|
| 29 |
-
use_auth_token=st.secrets["USER_TOKEN"]
|
| 30 |
).to(device)
|
| 31 |
|
| 32 |
pipe.safety_checker = None
|
|
|
|
| 26 |
# init stable diffusion model
|
| 27 |
pipe = ComposableStableDiffusionPipeline.from_pretrained(
|
| 28 |
"CompVis/stable-diffusion-v1-4",
|
|
|
|
| 29 |
).to(device)
|
| 30 |
|
| 31 |
pipe.safety_checker = None
|