danavirtual commited on
Commit
01ca166
Β·
1 Parent(s): 3abff73

requirements for ext class

Browse files
InstructionTextGenerationPipeline.py CHANGED
@@ -1,3 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  class InstructionTextGenerationPipeline:
2
  def __init__(
3
  self,
 
1
+ from torch import *
2
+ import gradio as gr
3
+ import requests
4
+ import torch
5
+ import transformers
6
+ import einops
7
+ ###
8
+ from typing import Any, Dict, Tuple
9
+ import warnings
10
+ import datetime
11
+ import os
12
+ from threading import Event, Thread
13
+ import torch
14
+ from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
15
+ import config
16
+
17
  class InstructionTextGenerationPipeline:
18
  def __init__(
19
  self,