use cogview3 model

app.py

@@ -5,25 +5,15 @@ import time
 from datetime import datetime, timedelta
 
 import gradio as gr
-import numpy as np
 import random
 import spaces  # [uncomment to use ZeroGPU]
-from diffusers import FluxPipeline
+from diffusers import CogView3PlusPipeline
 import torch
 from openai import OpenAI
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "black-forest-labs/FLUX.1-dev"
 
-if torch.cuda.is_available():
-    torch_dtype = torch.float16
-else:
-    torch_dtype = torch.float32
-
-pipe = FluxPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
-pipe = pipe.to(device)
-
-MAX_SEED = np.iinfo(np.int32).max
+pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3-Plus-3B", torch_dtype=torch.bfloat16).to(device)
 
 
 def clean_string(s):
@@ -134,9 +124,7 @@ threading.Thread(target=delete_old_files, daemon=True).start()
 def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
           progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-
-    generator = torch.Generator().manual_seed(seed)
+        seed = random.randint(0, 65536)
 
     image = pipe(
         prompt=prompt,
@@ -145,7 +133,7 @@ def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_infer
         num_inference_steps=num_inference_steps,
         width=width,
         height=height,
-        generator=generator
+        generator=torch.Generator().manual_seed(seed)
     ).images[0]
     return image, seed
 
@@ -207,7 +195,7 @@ with gr.Blocks(css=css) as demo:
             seed = gr.Slider(
                 label="Seed",
                 minimum=0,
-                maximum=MAX_SEED,
+                maximum=65536,
                 step=1,
                 value=0,
             )
@@ -220,7 +208,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=512,
                     maximum=2048,
                     step=32,
-                    value=1024,  # 替换为你的模型默认值
+                    value=1024,
                 )
 
                 height = gr.Slider(