jadechoghari commited on
Commit
f231447
1 Parent(s): 6d56da9

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from PIL import Image
3
  from conversation import conv_templates
4
- from builder import load_pretrained_model # Assuming this is your custom model loader
5
  from functools import partial
6
  from typing import Optional, Callable
7
  import ast
 
1
  import torch
2
  from PIL import Image
3
  from conversation import conv_templates
4
+ from builder import load_pretrained_model
5
  from functools import partial
6
  from typing import Optional, Callable
7
  import ast