Я не могу устранить ошибку, связанную с тем, что model.safetensor не найден при выполнении моего кода?

Question

Я не могу устранить ошибку в своем коде, которая заключается в создании изображения с помощью текста, и он продолжает выдавать ошибку, что model.safetensor не найден или unet\diffusion_pytorch_model.safetensors не найден, и я пытался загрузить эти файлы и попытался поместите его в несколько каталогов, где он мог бы его использовать, но тогда он также не будет исполняемым.

       from auth_token import auth_token
    from fastapi import FastAPI, Response
    from fastapi.middleware.cors import CORSMiddleware
    import torch
    from torch import autocast
    from diffusers import StableDiffusionPipeline
    from io import BytesIO
    import base64 
    
    
    app = FastAPI()
    
    app.add_middleware(
        CORSMiddleware, 
        allow_credentials=True, 
        allow_origins=["*"], 
        allow_methods=["*"], 
        allow_headers=["*"]
    )
    
    device = "cuda"
    model_id = "CompVis/stable-diffusion-v1-4"
    pipe = StableDiffusionPipeline.from_pretrained(model_id, revision="fp16", torch_dtype=torch.float16, use_auth_token=auth_token)
    pipe.to(device)
    
    @app.get("/")
    def generate(prompt: str): 
        with autocast(device): 
            image = pipe(prompt, guidance_scale=8.5).images[0]
    
        image.save("testimage.png")
        buffer = BytesIO()
        image.save(buffer, format="PNG")
        imgstr = base64.b64encode(buffer.getvalue())
    
        return Response(content=imgstr, media_type="image/png")