Spaces:
Runtime error
Runtime error
disable fp16
Browse files
app.py
CHANGED
@@ -28,10 +28,7 @@ if 'pipeline' not in st.session_state:
|
|
28 |
if model != st.session_state.model or st.session_state.pipeline is None:
|
29 |
if model:
|
30 |
with st.spinner("Loading Model..."):
|
31 |
-
pipeline = StableDiffusionPipeline.from_pretrained(
|
32 |
-
model,
|
33 |
-
torch_dtype=torch.float16
|
34 |
-
)
|
35 |
assert type(pipeline) is StableDiffusionPipeline
|
36 |
if torch.cuda.is_available():
|
37 |
pipeline = pipeline.to("cuda")
|
|
|
28 |
if model != st.session_state.model or st.session_state.pipeline is None:
|
29 |
if model:
|
30 |
with st.spinner("Loading Model..."):
|
31 |
+
pipeline = StableDiffusionPipeline.from_pretrained(model)
|
|
|
|
|
|
|
32 |
assert type(pipeline) is StableDiffusionPipeline
|
33 |
if torch.cuda.is_available():
|
34 |
pipeline = pipeline.to("cuda")
|