KaizeShi commited on
Commit
91df19c
·
1 Parent(s): 73cbc9a

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import torch
2
  from peft import PeftModel
3
  import transformers
4
  import gradio as gr
 
5
 
6
  assert (
7
  "LlamaTokenizer" in transformers._import_structure["models.llama"]
 
2
  from peft import PeftModel
3
  import transformers
4
  import gradio as gr
5
+ import os
6
 
7
  assert (
8
  "LlamaTokenizer" in transformers._import_structure["models.llama"]