Hugging Face's logo

Spaces:
snyderline
/
Lumina-Image-2.0
Running on Zero

App Files Community
benjamin-paine commited on
Commit
d28d8ca
·
verified ·
1 Parent(s): eb226af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ with safe_open(model_file, framework="pt") as f:
51
 
52
  state_dict = convert_sd3_transformer_checkpoint_to_diffusers(state_dict)
53
  transformer.load_state_dict(state_dict)
54
- transformer.to_empty(device)
55
 
56
  # Try to keep memory usage down
57
  del state_dict
 
51
 
52
  state_dict = convert_sd3_transformer_checkpoint_to_diffusers(state_dict)
53
  transformer.load_state_dict(state_dict)
54
+ transformer.to_empty(device=device)
55
 
56
  # Try to keep memory usage down
57
  del state_dict