Commit
•
99cd496
1
Parent(s):
81c54f7
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -72,7 +72,7 @@ class MARModel(DiffusionPipeline):
|
|
| 72 |
).cuda()
|
| 73 |
|
| 74 |
# use safetensors
|
| 75 |
-
state_dict = load_file(
|
| 76 |
model.load_state_dict(state_dict)
|
| 77 |
model.eval()
|
| 78 |
|
|
|
|
| 72 |
).cuda()
|
| 73 |
|
| 74 |
# use safetensors
|
| 75 |
+
state_dict = load_file(model_checkpoint_path)
|
| 76 |
model.load_state_dict(state_dict)
|
| 77 |
model.eval()
|
| 78 |
|