Hugging Face's logo

Spaces:
rasyosef
/
phi-1_5-chat
Running

App Files Community
rasyosef commited on
Commit
22bc9c5
•
1 Parent(s): b6cdfb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ checkpoint = "rasyosef/Phi-1_5-Instruct-v0.1"
9
 
10
  # Download and load model and tokenizer
11
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
- model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float32, device_map="auto")
13
 
14
  # Text generation pipeline
15
  phi1_5 = pipeline(
 
9
 
10
  # Download and load model and tokenizer
11
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
12
+ model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float32, device_map="cpu")
13
 
14
  # Text generation pipeline
15
  phi1_5 = pipeline(