Hugging Face's logo

Spaces:
pyresearch
/
Fake-news-facts
Sleeping

App Files Community
pyresearch commited on
Commit
86a2049
·
verified ·
1 Parent(s): ac5daff

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
 
10
 
11
 
12
- torch.set_default_device("cpu")
13
 
14
  # Load the 'microsoft/phi-2' model and tokenizer
15
  model_phi2 = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype="auto", trust_remote_code=True)
 
9
 
10
 
11
 
12
+ torch.set_default_device("cuda")
13
 
14
  # Load the 'microsoft/phi-2' model and tokenizer
15
  model_phi2 = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype="auto", trust_remote_code=True)