Hugging Face's logo

Spaces:
Bofeee5675
/
FIRE
Running

App Files Community
zhangbofei commited on
Commit
a22404f
•
1 Parent(s): 10867a7

fix: import

Browse files
Files changed (2) hide show
  1. src/model/model_falcon.py +1 -1
  2. src/model/model_yuan2.py +1 -1
src/model/model_falcon.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  import transformers
7
  from transformers import TextIteratorStreamer, GenerationConfig
8
 
9
- from fastchat.utils import is_partial_stop
10
 
11
 
12
  @torch.inference_mode()
 
6
  import transformers
7
  from transformers import TextIteratorStreamer, GenerationConfig
8
 
9
+ from src.utils import is_partial_stop
10
 
11
 
12
  @torch.inference_mode()
src/model/model_yuan2.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  import transformers
7
  from transformers import TextIteratorStreamer, GenerationConfig
8
 
9
- from fastchat.utils import is_partial_stop
10
 
11
 
12
  @torch.inference_mode()
 
6
  import transformers
7
  from transformers import TextIteratorStreamer, GenerationConfig
8
 
9
+ from src.utils import is_partial_stop
10
 
11
 
12
  @torch.inference_mode()