Spaces:
Sleeping
Sleeping
added new class
Browse files- 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):
|