reab5555 commited on
Commit
f8b2925
·
verified ·
1 Parent(s): 1fa3901

Update processing.py

Browse files
Files changed (1) hide show
  1. processing.py +1 -0
processing.py CHANGED
@@ -52,6 +52,7 @@ class LazyPipeline:
52
  top_p = 0.9,
53
  top_k = 50,
54
  repetition_penalty = 1.2,
 
55
  )
56
  return self.pipeline
57
 
 
52
  top_p = 0.9,
53
  top_k = 50,
54
  repetition_penalty = 1.2,
55
+ device_map="auto",
56
  )
57
  return self.pipeline
58