hanzla commited on
Commit
55d1776
·
1 Parent(s): 0413f13

sliders added

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -28,9 +28,8 @@ model_llm = AutoModelForCausalLM.from_pretrained(
28
  device_map="cuda",
29
  torch_dtype="auto",
30
  trust_remote_code=True,
31
- device="cuda"
32
  )
33
- tokenizer_llm = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct",device="cuda")
34
 
35
  pipe = AnimateDiffPipeline.from_pretrained(model_id, motion_adapter=adapter, torch_dtype=torch.float16).to(device)
36
  scheduler = DDIMScheduler.from_pretrained(
@@ -56,9 +55,8 @@ def generate_video(prompt,negative_prompt, guidance_scale, num_inference_steps,
56
  "text-generation",
57
  model=model_llm,
58
  tokenizer=tokenizer_llm,
59
- device="cuda"
60
  )
61
-
62
  generation_args = {
63
  "max_new_tokens": 500,
64
  "return_full_text": False,
 
28
  device_map="cuda",
29
  torch_dtype="auto",
30
  trust_remote_code=True,
 
31
  )
32
+ tokenizer_llm = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct")
33
 
34
  pipe = AnimateDiffPipeline.from_pretrained(model_id, motion_adapter=adapter, torch_dtype=torch.float16).to(device)
35
  scheduler = DDIMScheduler.from_pretrained(
 
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,