Skip to content

Commit

Permalink
Update ios/RnExecutorch/models/BaseModel.mm
Browse files Browse the repository at this point in the history
Co-authored-by: mkopcins <[email protected]>
  • Loading branch information
NorbertKlockiewicz and mkopcins authored Dec 9, 2024
1 parent c6b8fe3 commit 3444d18
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions ios/RnExecutorch/models/BaseModel.mm
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@ - (NSArray *)forward:(NSArray *)input shape:(NSArray *)shape inputType:(NSNumber
- (void)loadModel:(NSURL *)modelURL completion:(void (^)(BOOL success, NSNumber* code))completion {
module = [[ETModel alloc] init];
[Fetcher fetchResource:modelURL resourceType:ResourceType::MODEL completionHandler:^(NSString *filePath, NSError *error) {
if (error) {
completion(NO, @-1);
return;
}
if (error) {
completion(NO, @-1);
return;
}
NSNumber *result = [self->module loadModel: filePath];
if(result != 0){
completion(NO, result);
Expand Down

0 comments on commit 3444d18

Please sign in to comment.