Futuretop commited on
Commit
00ca8e1
·
verified ·
1 Parent(s): 5d40972

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -35,8 +35,7 @@ pipe = DiffusionPipeline.from_pretrained(
35
  ),
36
  tokenizer=tokenizer,
37
  feature_extractor=feature_extractor,
38
- torch_dtype=torch.float16, # load weights in half-precision
39
- revision="fp16", # if your repo provides fp16 weights
40
  use_safetensors=True
41
  )
42
 
 
35
  ),
36
  tokenizer=tokenizer,
37
  feature_extractor=feature_extractor,
38
+ torch_dtype=torch.float32, # load weights in half-precision
 
39
  use_safetensors=True
40
  )
41