RandomNameAnd6 commited on
Commit
8a50e1a
·
verified ·
1 Parent(s): e486f7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -7,10 +7,7 @@ import torch
7
  import random
8
  from transformers import AutoTokenizer, AutoModelForCausalLM
9
 
10
- max_seq_length = 2048 # Choose any! We auto support RoPE Scaling internally!
11
- dtype = None # None for auto detection. Float16 for Tesla T4, V100, Bfloat16 for Ampere+
12
-
13
- model, tokenizer = AutoModelForCausalLM.from_pretrained("unsloth/Qwen2-1.5B-bnb-4bit", max_seq_length = max_seq_length, dtype = dtype)
14
  model = PeftModel.from_pretrained(model, "RandomNameAnd6/Phi-3-Mini-Dhar-Mann-Adapters-BOS")
15
 
16
  def generate_text(prompt):
 
7
  import random
8
  from transformers import AutoTokenizer, AutoModelForCausalLM
9
 
10
+ model, tokenizer = AutoModelForCausalLM.from_pretrained("unsloth/Qwen2-1.5B-bnb-4bit")
 
 
 
11
  model = PeftModel.from_pretrained(model, "RandomNameAnd6/Phi-3-Mini-Dhar-Mann-Adapters-BOS")
12
 
13
  def generate_text(prompt):