keminglu commited on
Commit
45f49e9
·
1 Parent(s): 1d3c260

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -1,3 +1,13 @@
1
  import gradio as gr
 
 
 
2
 
3
- gr.Interface.load("models/keminglu/pivoine-7b", api_key="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").launch()
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import torch
3
+ import json
4
+ from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
+ tokenizer = AutoTokenizer.from_pretrained("keminglu/pivoine-7b", use_auth_token="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", padding_side="left")
7
+ model = AutoModelForCausalLM.from_pretrained("keminglu/pivoine-7b", use_auth_token="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", torch_dtype=torch.float16)
8
+ #input_device = torch.device("cuda:5")
9
+ model.requires_grad_(False)
10
+ model.eval()
11
+ #model = model.to(input_device)
12
+
13
+ # examples = json.load(open("examples.json"))