liiablack commited on
Commit
386db34
·
verified ·
1 Parent(s): 136a865

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def pretty_features(row: dict) -> dict:
59
  from diffusers import DiffusionPipeline
60
 
61
  device = "cpu"
62
- pipe = DiffusionPipeline.from_pretrained(MODEL_ID, dtype=torch.float32)
63
  pipe.to(device)
64
  if hasattr(pipe, "enable_attention_slicing"):
65
  pipe.enable_attention_slicing()
 
59
  from diffusers import DiffusionPipeline
60
 
61
  device = "cpu"
62
+ pipe = DiffusionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float32)
63
  pipe.to(device)
64
  if hasattr(pipe, "enable_attention_slicing"):
65
  pipe.enable_attention_slicing()