Commit
•
81c54f7
1
Parent(s):
7c91606
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_path)
|
| 76 |
model.load_state_dict(state_dict)
|
| 77 |
model.eval()
|
| 78 |
|