Commit 20c74e57 authored by mashun1's avatar mashun1
Browse files

fix bug

parent 2103dd66
// !!! This is a file automatically generated by hipify!!! // !!! This is a file automatically generated by hipify!!!
#include <ATen/dtk_macros.h>
#include "hip/hip_runtime.h" #include "hip/hip_runtime.h"
/*! /*!
******************* BEGIN Caffe Copyright Notice and Disclaimer **************** ******************* BEGIN Caffe Copyright Notice and Disclaimer ****************
......
// !!! This is a file automatically generated by hipify!!! // !!! This is a file automatically generated by hipify!!!
#include <ATen/dtk_macros.h>
// from https://github.com/rosinality/stylegan2-pytorch/blob/master/op/fused_bias_act_kernel.cu // from https://github.com/rosinality/stylegan2-pytorch/blob/master/op/fused_bias_act_kernel.cu
// Copyright (c) 2019, NVIDIA Corporation. All rights reserved. // Copyright (c) 2019, NVIDIA Corporation. All rights reserved.
// //
......
// !!! This is a file automatically generated by hipify!!! // !!! This is a file automatically generated by hipify!!!
#include <ATen/dtk_macros.h>
// from https://github.com/rosinality/stylegan2-pytorch/blob/master/op/upfirdn2d_kernel.cu // from https://github.com/rosinality/stylegan2-pytorch/blob/master/op/upfirdn2d_kernel.cu
// Copyright (c) 2019, NVIDIA Corporation. All rights reserved. // Copyright (c) 2019, NVIDIA Corporation. All rights reserved.
// //
......
...@@ -9,12 +9,13 @@ from typing import Optional, Any ...@@ -9,12 +9,13 @@ from typing import Optional, Any
from ldm.modules.diffusionmodules.util import checkpoint from ldm.modules.diffusionmodules.util import checkpoint
try: # try:
import xformers # import xformers
import xformers.ops # import xformers.ops
XFORMERS_IS_AVAILBLE = True # XFORMERS_IS_AVAILBLE = True
except: # except:
XFORMERS_IS_AVAILBLE = False # XFORMERS_IS_AVAILBLE = False
XFORMERS_IS_AVAILBLE = False
# CrossAttn precision handling # CrossAttn precision handling
import os import os
......
...@@ -8,13 +8,16 @@ from typing import Optional, Any ...@@ -8,13 +8,16 @@ from typing import Optional, Any
from ldm.modules.attention import MemoryEfficientCrossAttention from ldm.modules.attention import MemoryEfficientCrossAttention
try: # try:
import xformers # import xformers
import xformers.ops # import xformers.ops
XFORMERS_IS_AVAILBLE = True # XFORMERS_IS_AVAILBLE = True
except: # except:
XFORMERS_IS_AVAILBLE = False # XFORMERS_IS_AVAILBLE = False
print("No module 'xformers'. Proceeding without it.") # print("No module 'xformers'. Proceeding without it.")
XFORMERS_IS_AVAILBLE = False
# print("No support 'xformers'. Proceeding without it.")
def get_timestep_embedding(timesteps, embedding_dim): def get_timestep_embedding(timesteps, embedding_dim):
......
test_results/test.png

516 KB | W: | H:

test_results/test.png

523 KB | W: | H:

test_results/test.png
test_results/test.png
test_results/test.png
test_results/test.png
  • 2-up
  • Swipe
  • Onion skin
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