ovis-image-7b-cpu

Running

smartdigitalnetworks commited on 2 days ago

Commit

7c271a9

verified ·

1 Parent(s): bc7bbfb

Update app_diffusers.py

Files changed (1) hide show

app_diffusers.py CHANGED Viewed

@@ -25,7 +25,7 @@ pipe = OvisImagePipeline.from_pretrained(
     token=hf_token,
     torch_dtype=torch.bfloat16
 )
-pipe.to("cuda")
 @spaces.GPU(duration=75)
 def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):

     token=hf_token,
     torch_dtype=torch.bfloat16
 )
+pipe.to(device)
 @spaces.GPU(duration=75)
 def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):