PEFT
English
medical
Tonic commited on
Commit
75687e5
1 Parent(s): 933497b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -72,8 +72,10 @@ import torch
72
  import gradio as gr
73
  import random
74
  from textwrap import wrap
75
- import random
76
-
 
 
77
  # Functions to Wrap the Prompt Correctly
78
 
79
  def wrap_text(text, width=90):
 
72
  import gradio as gr
73
  import random
74
  from textwrap import wrap
75
+ from transformers import AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
76
+ from peft import PeftModel, PeftConfig
77
+ import torch
78
+ import gradio as gr
79
  # Functions to Wrap the Prompt Correctly
80
 
81
  def wrap_text(text, width=90):