Spaces:
Runtime error
Runtime error
Update processing.py
Browse files- 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 |
|