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')
|