Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,6 +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 |
|
| 55 |
# Try to keep memory usage down
|
| 56 |
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)
|
| 55 |
|
| 56 |
# Try to keep memory usage down
|
| 57 |
del state_dict
|