MergeLlama-7b

Paused

codys12 commited on Oct 17, 2023

Commit

192a9de

•

1 Parent(s): b3c3af3

import peft

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,6 +8,7 @@ import gradio as gr
 import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 DESCRIPTION = "# Mistral-7B"

 import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+from peft import PeftModel, PeftConfig
 DESCRIPTION = "# Mistral-7B"