helloperson123 commited on
Commit
1ffd59d
·
verified ·
1 Parent(s): ab911fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -11,13 +11,15 @@ CORS(app)
11
  # Model (UNGATED, FREE)
12
  # ----------------------------
13
  MODEL_NAME = "microsoft/phi-3-mini-4k-instruct"
14
- DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
 
16
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
17
  model = AutoModelForCausalLM.from_pretrained(
18
  MODEL_NAME,
19
- torch_dtype=torch.float16 if DEVICE == "cuda" else torch.float32
20
- ).to(DEVICE)
 
 
 
21
 
22
  # ----------------------------
23
  # SYSTEM PROMPT
 
11
  # Model (UNGATED, FREE)
12
  # ----------------------------
13
  MODEL_NAME = "microsoft/phi-3-mini-4k-instruct"
14
+ DEVICE = "cpu"
15
 
 
16
  model = AutoModelForCausalLM.from_pretrained(
17
  MODEL_NAME,
18
+ torch_dtype=torch.float32
19
+ )
20
+
21
+ model.to(DEVICE)
22
+ )
23
 
24
  # ----------------------------
25
  # SYSTEM PROMPT