YangWu001 commited on
Commit
ecf9cdb
·
1 Parent(s): 7aa7dac
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
 
6
  # Inference client setup
7
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
- # pipe = pipeline("text-generation", "microsoft/Phi-3-mini-4k-instruct", torch_dtype=torch.bfloat16, device_map="auto")
9
 
10
  # Global flag to handle cancellation
11
  stop_inference = False
 
5
 
6
  # Inference client setup
7
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
+ pipe = pipeline("text-generation", "microsoft/Phi-3-mini-4k-instruct", torch_dtype=torch.bfloat16, device_map="auto")
9
 
10
  # Global flag to handle cancellation
11
  stop_inference = False