Spaces:
Runtime error
Runtime error
sliders added
Browse files
app.py
CHANGED
|
@@ -55,8 +55,8 @@ def generate_video(prompt,negative_prompt, guidance_scale, num_inference_steps,
|
|
| 55 |
"text-generation",
|
| 56 |
model=model_llm,
|
| 57 |
tokenizer=tokenizer_llm,
|
|
|
|
| 58 |
)
|
| 59 |
-
pipe_llm.to(device)
|
| 60 |
generation_args = {
|
| 61 |
"max_new_tokens": 500,
|
| 62 |
"return_full_text": False,
|
|
|
|
| 55 |
"text-generation",
|
| 56 |
model=model_llm,
|
| 57 |
tokenizer=tokenizer_llm,
|
| 58 |
+
device_map='auto'
|
| 59 |
)
|
|
|
|
| 60 |
generation_args = {
|
| 61 |
"max_new_tokens": 500,
|
| 62 |
"return_full_text": False,
|