gufett0 commited on
Commit
5794390
·
1 Parent(s): f797fbc

added new class

Browse files
Files changed (1) hide show
  1. interface.py +1 -0
interface.py CHANGED
@@ -5,6 +5,7 @@ from typing import Any
5
  import torch
6
  from transformers import TextIteratorStreamer
7
  from threading import Thread
 
8
 
9
  # for transformers 2
10
  class GemmaLLMInterface(CustomLLM):
 
5
  import torch
6
  from transformers import TextIteratorStreamer
7
  from threading import Thread
8
+ from pydantic import Field
9
 
10
  # for transformers 2
11
  class GemmaLLMInterface(CustomLLM):