mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 08:21:53 +08:00
[BugFix] Fix model loading error for 300B FP8 EP parallel test case (#6382)
* fix fp8 bug * fix * fix comment, cn to en * fix ci * del else in utils * fix review
This commit is contained in:
@@ -58,7 +58,7 @@ class DefaultModelLoaderV1(BaseModelLoader):
|
||||
load_weights_from_cache(model, weights_iterator)
|
||||
else:
|
||||
model.load_weights(weights_iterator)
|
||||
|
||||
# Execute post-processing after weight loading
|
||||
process_final_after_loading(model, fd_config)
|
||||
|
||||
self.clean_memory_fragments()
|
||||
|
||||
Reference in New Issue
Block a user