cleanup and local test
This commit is contained in:
committed by
SolitaryThinker
parent
2d7e262b13
commit
a0749e98fd
@@ -1,13 +1,4 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
"""
|
||||
GLM-Image inference example using FastVideo.
|
||||
|
||||
This example demonstrates how to generate images using the GLM-Image model
|
||||
through FastVideo's VideoGenerator API.
|
||||
|
||||
Usage:
|
||||
python examples/inference/basic/basic_glm_image.py
|
||||
"""
|
||||
|
||||
import os
|
||||
|
||||
@@ -41,11 +32,12 @@ def main() -> None:
|
||||
height=1024,
|
||||
width=1024,
|
||||
num_inference_steps=50,
|
||||
guidance_scale=7.5,
|
||||
guidance_scale=1.5,
|
||||
)
|
||||
|
||||
if result and len(result) > 0:
|
||||
img = Image.fromarray(result[0])
|
||||
frames = result.get("frames") if isinstance(result, dict) else None
|
||||
if frames:
|
||||
img = Image.fromarray(frames[0])
|
||||
img.save(os.path.join(OUTPUT_PATH, "landscape.png"))
|
||||
print(f"Saved image to {OUTPUT_PATH}/landscape.png")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user