Skip to content

H
Hydra Node Http
Commit 7dc1b8b3 authored by novelailab's avatar novelailab
Browse files
Options

Revert determinism

parent a7c71667
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
hydra_node/config.py
View file @ 7dc1b8b3
import os import os
os.environ["CUBLAS_WORKSPACE_CONFIG"]=":4096:8"
import torch import torch
torch.use_deterministic_algorithms(True, warn_only=True)
import logging import logging
import os import os
import platform import platform
......
main.py
View file @ 7dc1b8b3
import os
os.environ["CUBLAS_WORKSPACE_CONFIG"]=":4096:8"
from fastapi import FastAPI, Request, Depends from fastapi import FastAPI, Request, Depends
from pydantic import BaseModel from pydantic import BaseModel
from fastapi.responses import HTMLResponse, PlainTextResponse, Response from fastapi.responses import HTMLResponse, PlainTextResponse, Response
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment