suwesh commited on
Commit
3bfeabd
Β·
verified Β·
1 Parent(s): df61aa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -12,9 +12,7 @@ modelpath = "distilgpt2"
12
 
13
  pipe = pipeline(
14
  "text-generation",
15
- model=modelpath,
16
- torch_dtype=torch.bfloat16,
17
- device_map="auto",
18
  )
19
  messages = [
20
  {"role": "system", "content": "You are a customer applying for a loan. Provide dummy details about your application and negotiate the terms."},
 
12
 
13
  pipe = pipeline(
14
  "text-generation",
15
+ model=modelpath
 
 
16
  )
17
  messages = [
18
  {"role": "system", "content": "You are a customer applying for a loan. Provide dummy details about your application and negotiate the terms."},