wybxc commited on
Commit
f9e11a1
1 Parent(s): a160296

disable fp16

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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")