Spaces:

AIDC-AI
/

Ovis-Image-7B

Running on Zero

Flourish commited on 8 days ago

Commit

c27ae96

verified ·

1 Parent(s): f01b653

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ ovis_encoder = OvisEmbedder(
 def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
     print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
     image = generate_image(
-        device=ovis_image.device,
         dtype=_dtype,
         model=ovis_image,
         prompt=prompt,

 def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
     print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
     image = generate_image(
+        device=next(ovis_image.parameters()).device,
         dtype=_dtype,
         model=ovis_image,
         prompt=prompt,