JeonghyeokDo/GeoSET main
imports osPython comment contains a URL
SHA-25679764311ce659d52d74d73dffff3e59b413045daf469d9862ffcceca99a5a7ca
MaleculeH(Cm)Md(In)
Evidence
1# Copyright (c) Black Forest Labs.
2# Copyright (c) 2026 The GeoSET Authors.
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8# http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under …
12:46… D, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15#
16# Adapted from black-forest-labs/flux2 (Apache-2.0). Modifie …
20:92… h the final velocity head.
21"""
22
23from __future__ import annotations
24
25import math
26import os
27
28import torch
29from diffusers.configuration_utils import ConfigMixin, register_to_config
30from diffusers.models.modeling_utils import ModelMixin
31from einops import rearrange
32from safetensors.torch import load_file
33from torch import Tensor, nn
34from torch.nn import functional as F
35
36
37def timestep_embedding(t: Tensor, dim: int, max_period: int = 10000, time_factor: float …