Spaces:
Runtime error
Runtime error
Commit ·
74d8330
1
Parent(s): de7f4e9
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ device = "cpu"
|
|
| 25 |
st.info('1')
|
| 26 |
#___________________________________________________________________________________________________________________________
|
| 27 |
|
| 28 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token='
|
| 29 |
def dummy(images, **kwargs): return images, False
|
| 30 |
pipe.safety_checker = dummy
|
| 31 |
st.info('2')
|
|
|
|
| 25 |
st.info('1')
|
| 26 |
#___________________________________________________________________________________________________________________________
|
| 27 |
|
| 28 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token='Add Your Info', torch_dtype=torch.float32)
|
| 29 |
def dummy(images, **kwargs): return images, False
|
| 30 |
pipe.safety_checker = dummy
|
| 31 |
st.info('2')
|