Caslow commited on
Commit
4427f62
·
1 Parent(s): 70bd322

revise requirement.txt

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -1,7 +1,7 @@
1
- from unsloth import FastLanguageModel
2
  from transformers import TextStreamer
3
  from typing import Tuple, List, Dict
4
  import torch
 
5
 
6
  def load_model(
7
  model_name: str,
 
 
1
  from transformers import TextStreamer
2
  from typing import Tuple, List, Dict
3
  import torch
4
+ from unsloth import FastLanguageModel
5
 
6
  def load_model(
7
  model_name: str,