OnlyCheeini commited on
Commit
5d9e718
·
verified ·
1 Parent(s): 9c8270f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ h1 {
50
 
51
  # Load the tokenizer and model
52
  tokenizer = AutoTokenizer.from_pretrained("OnlyCheeini/greesychat-v3")
53
- model = AutoModelForCausalLM.from_pretrained("OnlyCheeini/greesychat-v3", device_map="auto") # to("cuda:0")
54
  terminators = [
55
  tokenizer.eos_token_id,
56
  tokenizer.convert_tokens_to_ids("<|eot_id|>")
 
50
 
51
  # Load the tokenizer and model
52
  tokenizer = AutoTokenizer.from_pretrained("OnlyCheeini/greesychat-v3")
53
+ model = AutoModelForCausalLM.from_pretrained("OnlyCheeini/greesychat-v3",offload_folder="offload", device_map="auto") # to("cuda:0")
54
  terminators = [
55
  tokenizer.eos_token_id,
56
  tokenizer.convert_tokens_to_ids("<|eot_id|>")